From 5ba50b5737c984e77bc20dc35adf5c5d78f22c48 Mon Sep 17 00:00:00 2001
From: chrishalcrow <57948917+chrishalcrow@users.noreply.github.com>
Date: Fri, 8 Nov 2024 09:56:36 +0000
Subject: [PATCH] oups

---
 src/spikeinterface/curation/tests/test_model_based_curation.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/spikeinterface/curation/tests/test_model_based_curation.py b/src/spikeinterface/curation/tests/test_model_based_curation.py
index 0ef6dc3cc9..436ef51e0e 100644
--- 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