Update src/spikeinterface/sortingcomponents/tools.py

Co-authored-by: Garcia Samuel <[email protected]>
SpikeInterface · Jan 14, 2025 · 827fb4d · 827fb4d
1 parent 23dcacd
commit 827fb4d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/spikeinterface/sortingcomponents/tools.py b/src/spikeinterface/sortingcomponents/tools.py
@@ -313,8 +313,8 @@ def get_shuffled_recording_slices(recording, seed=None, **job_kwargs):
         chunks = divide_segment_into_chunks(num_frames, chunk_size)
         recording_slices.extend([(segment_index, frame_start, frame_stop) for frame_start, frame_stop in chunks])
 
-    if seed is not None:
-        rng = np.random.RandomState(seed)
-        recording_slices = rng.permutation(recording_slices)
+
+    rng = np.random.default_rng(seed)
+    recording_slices = rng.permutation(recording_slices)
 
     return recording_slices