From 1fa0ee2a4e8dd1b7d01ee78f05d99750590bb0c2 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Fri, 21 Jun 2024 13:34:06 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- src/spikeinterface/curation/model_based_curation.py | 1 - src/spikeinterface/curation/tests/test_model_based_curation.py | 1 + 2 files changed, 1 insertion(+), 1 deletion(-) 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 ):