Skip to content

Commit 096d91a

Browse files
authored
Merge pull request #3388 from jiumao2/main
Fix integer overflow of the total sample number when concatenating long recordings
2 parents c240155 + ccc8a38 commit 096d91a

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

src/spikeinterface/core/segmentutils.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -157,7 +157,7 @@ def __init__(self, parent_segments, sampling_frequency, ignore_times=True):
157157
self.parent_segments = parent_segments
158158
self.all_length = [rec_seg.get_num_samples() for rec_seg in self.parent_segments]
159159
self.cumsum_length = [0] + [sum(self.all_length[: i + 1]) for i in range(len(self.all_length))]
160-
self.total_length = int(np.sum(self.all_length))
160+
self.total_length = int(sum(self.all_length))
161161

162162
def get_num_samples(self):
163163
return self.total_length
@@ -450,7 +450,7 @@ def __init__(self, parent_segments, parent_num_samples, sampling_frequency):
450450
self.parent_segments = parent_segments
451451
self.parent_num_samples = parent_num_samples
452452
self.cumsum_length = np.cumsum([0] + self.parent_num_samples)
453-
self.total_num_samples = np.sum(self.parent_num_samples)
453+
self.total_num_samples = int(sum(self.parent_num_samples))
454454

455455
def get_num_samples(self):
456456
return self.total_num_samples

0 commit comments

Comments
 (0)