Merge pull request #3426 from jiumao2/main

Fix integer overflow in parallel computing
SpikeInterface · Sep 23, 2024 · 57c4e2b · 57c4e2b
2 parents 50eadce + 41f4c31
commit 57c4e2b
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/src/spikeinterface/core/job_tools.py b/src/spikeinterface/core/job_tools.py
@@ -136,11 +136,8 @@ def divide_segment_into_chunks(num_frames, chunk_size):
     else:
         n = num_frames // chunk_size
 
-        frame_starts = np.arange(n) * chunk_size
-        frame_stops = frame_starts + chunk_size
-
-        frame_starts = frame_starts.tolist()
-        frame_stops = frame_stops.tolist()
+        frame_starts = [i * chunk_size for i in range(n)]
+        frame_stops = [frame_start + chunk_size for frame_start in frame_starts]
 
         if (num_frames % chunk_size) > 0:
             frame_starts.append(n * chunk_size)