Skip to content

Commit 540eee7

Browse files
committed
better fix
1 parent 420668d commit 540eee7

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

src/spikeinterface/core/recording_tools.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -571,7 +571,7 @@ def get_random_recording_slices(
571571
# check chunk size
572572
num_segments = recording.get_num_segments()
573573
for segment_index in range(num_segments):
574-
chunk_size_limit = recording.get_num_frames(segment_index) - 2 * margin_frames - 1
574+
chunk_size_limit = recording.get_num_frames(segment_index) - 2 * margin_frames
575575
if chunk_size > chunk_size_limit:
576576
chunk_size = chunk_size_limit - 1
577577
warnings.warn(
@@ -585,12 +585,13 @@ def get_random_recording_slices(
585585
size = num_chunks_per_segment
586586
for segment_index in range(num_segments):
587587
num_frames = recording.get_num_frames(segment_index)
588-
high = num_frames - chunk_size - margin_frames
588+
high = num_frames - chunk_size - margin_frames + 1
589589
random_starts = rng.integers(low=low, high=high, size=size)
590590
random_starts = np.sort(random_starts)
591591
recording_slices += [
592592
(segment_index, start_frame, (start_frame + chunk_size)) for start_frame in random_starts
593593
]
594+
print(recording_slices)
594595
else:
595596
raise ValueError(f"get_random_recording_slices : wrong method {method}")
596597

0 commit comments

Comments
 (0)