Update generate ground truth to generate recording

SpikeInterface · May 29, 2024 · 567ff64 · 567ff64
1 parent ed5f90c
commit 567ff64
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/spikeinterface/preprocessing/tests/test_motion.py b/src/spikeinterface/preprocessing/tests/test_motion.py
@@ -3,10 +3,9 @@
 
 import shutil
 
-from spikeinterface.core import generate_ground_truth_recording
-from spikeinterface.preprocessing import correct_motion, load_motion_info
+from spikeinterface.core import generate_recording
 
-from spikeinterface.extractors import toy_example
+from spikeinterface.preprocessing import correct_motion, load_motion_info
 
 import numpy as np
 
@@ -19,7 +18,7 @@
 
 
 def test_estimate_and_correct_motion():
-    rec, sorting = generate_ground_truth_recording(durations=[30.0], num_units=10, num_channels=12)
+    rec = generate_recording(durations=[30.0], num_channels=12)
 
     print(rec)