Skip to content

Commit

Permalink
resolving black version formatting v.2 πŸ˜”
Browse files Browse the repository at this point in the history
  • Loading branch information
Butsko Christina committed Aug 22, 2024
1 parent 753dc9a commit 4f40b1f
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions presto/eval.py
Original file line number Diff line number Diff line change
Expand Up @@ -108,10 +108,10 @@ def __init__(
# use classes obtained from train to trim val and test classes
self.val_df.loc[
~self.val_df[class_column].isin(train_classes), class_column
] = "other_crop"
] = "other_crop"
self.test_df.loc[
~self.test_df[class_column].isin(train_classes), class_column
] = "other_crop"
] = "other_crop"

# create one-hot representation from obtained labels
# one-hot is needed for finetuning,
Expand Down

0 comments on commit 4f40b1f

Please sign in to comment.