Skip to content

Commit

Permalink
Flake8 fix
Browse files Browse the repository at this point in the history
  • Loading branch information
MarcelRobeer committed May 12, 2021
1 parent 6dd88c4 commit 7fa0012
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion contrastive_explanation/domain_mappers.py
Original file line number Diff line number Diff line change
Expand Up @@ -308,7 +308,8 @@ def apply_encode(self, data):
if one_instance: # Single instance
data = data.reshape(1, -1)

x = [self.encoders[i].transform(column).toarray() if i in self.categorical_features
x = [self.encoders[i].transform(column).toarray()
if i in self.categorical_features
else column.reshape(-1, 1) for i, column in enumerate(data.T)]

if one_instance:
Expand Down

0 comments on commit 7fa0012

Please sign in to comment.