From 6aaf01f13547a86cfd6fb9e02322bfec0292f78c Mon Sep 17 00:00:00 2001 From: chrishalcrow <57948917+chrishalcrow@users.noreply.github.com> Date: Mon, 16 Dec 2024 14:36:59 +0000 Subject: [PATCH] Update so csv can use subset of metrics --- src/spikeinterface/curation/tests/test_train_manual_curation.py | 2 -- src/spikeinterface/curation/train_manual_curation.py | 2 +- 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/src/spikeinterface/curation/tests/test_train_manual_curation.py b/src/spikeinterface/curation/tests/test_train_manual_curation.py index 0cb74d6d58..0a3ca2d45b 100644 --- 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 index 23c45cf776..5f25d1e3e7 100644 --- 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."""