From f245b15282047698553c9eabb2a142e75c3f840a Mon Sep 17 00:00:00 2001
From: Roberto <37729096+RobertoDF@users.noreply.github.com>
Date: Thu, 6 Feb 2025 16:21:47 +0100
Subject: [PATCH] fix  plot_1_automated_curation.py

---
 examples/tutorials/curation/plot_1_automated_curation.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/tutorials/curation/plot_1_automated_curation.py b/examples/tutorials/curation/plot_1_automated_curation.py
index ae01353201..00bd606c44 100644
--- a/examples/tutorials/curation/plot_1_automated_curation.py
+++ b/examples/tutorials/curation/plot_1_automated_curation.py
@@ -79,7 +79,7 @@
 # We can check that this is true by accessing the extension data.
 
 all_metric_names = list(sorting_analyzer.get_extension('quality_metrics').get_data().keys()) + list(sorting_analyzer.get_extension('template_metrics').get_data().keys())
-print(set(all_metric_names) == set(model.feature_names_in_))
+print(set(model.feature_names_in_).issubset(set(all_metric_names)))
 
 ##############################################################################
 # Great! We can now use the model to predict labels. Here, we pass the HF repo id directly