oups

SpikeInterface · Nov 8, 2024 · 5ba50b5 · 5ba50b5
1 parent e99c3b9
commit 5ba50b5
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/spikeinterface/curation/tests/test_model_based_curation.py b/src/spikeinterface/curation/tests/test_model_based_curation.py
@@ -1,6 +1,6 @@
 import pytest
 from pathlib import Path
-from spikeinterface.curation.tests.common import make_sorting_analyzer
+from spikeinterface.curation.tests.common import make_sorting_analyzer, sorting_analyzer_for_curation
 from spikeinterface.curation.model_based_curation import ModelBasedClassification
 import numpy as np
 
@@ -18,7 +18,6 @@ def pipeline():
 
     # Load trained_pipeline.skops
     unknown_types = get_untrusted_types(file=pipeline_path)
-    print(unknown_types)
     pipeline = load(pipeline_path, trusted=unknown_types)
     return pipeline