Update so csv can use subset of metrics

SpikeInterface · Dec 16, 2024 · 6aaf01f · 6aaf01f
1 parent c6739df
commit 6aaf01f
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/src/spikeinterface/curation/tests/test_train_manual_curation.py b/src/spikeinterface/curation/tests/test_train_manual_curation.py
@@ -89,8 +89,6 @@ def test_apply_scaling_imputation(trainer):
     assert np.all(y_train == y_train_scaled)
     assert np.all(y_test == y_test_scaled)
 
-    print(type(scaler))
-
     assert isinstance(imputer, KNNImputer)
     assert isinstance(scaler, StandardScaler)
 

diff --git a/src/spikeinterface/curation/train_manual_curation.py b/src/spikeinterface/curation/train_manual_curation.py
@@ -369,7 +369,7 @@ def process_test_data_for_classification(self):
             raise KeyError(f"{str(e)}, metrics_list contains invalid metric names")
 
         self.X = self.testing_metrics.reindex(columns=self.metric_names)
-        self.X = _format_metric_dataframe(self.testing_metrics)
+        self.X = _format_metric_dataframe(self.X)
 
     def apply_scaling_imputation(self, imputation_strategy, scaling_technique, X_train, X_test, y_train, y_test):
         """Impute and scale the data using the specified techniques."""