Skip to content

Commit 5ba73db

Browse files
authored
Merge pull request #3674 from RobertoDF/patch-1
Fix plot_1_automated_curation.py
2 parents b22fc37 + f245b15 commit 5ba73db

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

examples/tutorials/curation/plot_1_automated_curation.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -79,7 +79,7 @@
7979
# We can check that this is true by accessing the extension data.
8080

8181
all_metric_names = list(sorting_analyzer.get_extension('quality_metrics').get_data().keys()) + list(sorting_analyzer.get_extension('template_metrics').get_data().keys())
82-
print(set(all_metric_names) == set(model.feature_names_in_))
82+
print(set(model.feature_names_in_).issubset(set(all_metric_names)))
8383

8484
##############################################################################
8585
# Great! We can now use the model to predict labels. Here, we pass the HF repo id directly

0 commit comments

Comments
 (0)