bye bye spikeextractors

catalystneuro · Jun 13, 2024 · bd801cc · bd801cc
1 parent 77ab285
commit bd801cc
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 9 deletions.
diff --git a/requirements-testing.txt b/requirements-testing.txt
@@ -1,4 +1,4 @@
 pytest
 pytest-cov
 parameterized==0.8.1
-spikeextractors>=0.9.10
+spikeinterface>=0.100.7
diff --git a/src/roiextractors/example_datasets/toy_example.py b/src/roiextractors/example_datasets/toy_example.py
@@ -157,14 +157,10 @@ def toy_example(
         mode=mode,
     )
 
-    # generate spike trains
-    import spikeextractors as se
-
-    rec, sort = se.example_datasets.toy_example(
-        duration=duration,
-        K=num_rois,
-        num_channels=1,
-        sampling_frequency=sampling_frequency,
+    from spikeinterface.core import generate_ground_truth_recording
+
+    rec, sort = generate_ground_truth_recording(
+        num_channels=1, durations=[duration], num_units=num_rois, sampling_frequency=sampling_frequency
     )
 
     # create decaying response