Skip to content

Commit

Permalink
Update job_tools.py
Browse files Browse the repository at this point in the history
  • Loading branch information
jiumao2 committed Sep 18, 2024
1 parent 4ab6223 commit b63221e
Showing 1 changed file with 2 additions and 5 deletions.
7 changes: 2 additions & 5 deletions src/spikeinterface/core/job_tools.py
Original file line number Diff line number Diff line change
Expand Up @@ -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 = [(i+1) * chunk_size for i in range(n)]

if (num_frames % chunk_size) > 0:
frame_starts.append(n * chunk_size)
Expand Down

0 comments on commit b63221e

Please sign in to comment.