diff --git a/src/spikeinterface/curation/model_based_curation.py b/src/spikeinterface/curation/model_based_curation.py index 6341e18282..ea905cbd9e 100644 --- a/src/spikeinterface/curation/model_based_curation.py +++ b/src/spikeinterface/curation/model_based_curation.py @@ -46,7 +46,6 @@ def __init__(self, sorting_analyzer: SortingAnalyzer, pipeline): self.pipeline = pipeline self.required_metrics = pipeline.feature_names_in_ - def predict_labels(self): """ Predicts the labels for the spike sorting data using the trained model. diff --git a/src/spikeinterface/curation/tests/test_model_based_curation.py b/src/spikeinterface/curation/tests/test_model_based_curation.py index ab0213e08e..06509a35da 100644 --- a/src/spikeinterface/curation/tests/test_model_based_curation.py +++ b/src/spikeinterface/curation/tests/test_model_based_curation.py @@ -41,6 +41,7 @@ def test_model_based_classification_init(sorting_analyzer_for_curation, pipeline assert model_based_classification.sorting_analyzer == sorting_analyzer_for_curation assert model_based_classification.pipeline == pipeline + def test_model_based_classification_get_metrics_for_classification( sorting_analyzer_for_curation, pipeline, required_metrics ):