Skip to content

Commit 584c2ae

Browse files
authored
Merge pull request #3230 from JoeZiminski/handle_none_in_noise_generator
Add check for None in 'NoiseGeneratorRecordingSegment' get_traces().
2 parents a8853d3 + 17736ad commit 584c2ae

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

src/spikeinterface/core/generate.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1220,6 +1220,12 @@ def get_traces(
12201220
end_frame: int | None = None,
12211221
channel_indices: list | None = None,
12221222
) -> np.ndarray:
1223+
1224+
if start_frame is None:
1225+
start_frame = 0
1226+
if end_frame is None:
1227+
end_frame = self.get_num_samples()
1228+
12231229
start_frame_within_block = start_frame % self.noise_block_size
12241230
end_frame_within_block = end_frame % self.noise_block_size
12251231
num_samples = end_frame - start_frame

0 commit comments

Comments
 (0)