Skip to content

Commit f245b15

Browse files
authored
fix plot_1_automated_curation.py
1 parent b22fc37 commit f245b15

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)