Merge pull request #2854 from h-mayorquin/fix_length_numpy_scalar_in_…

…margin

Possible fix for #2837

src/spikeinterface/core/recording_tools.py

@@ -709,7 +709,7 @@ def get_chunk_with_margin(
     case zero padding is used, in the second case np.pad is called
     with mod="reflect".
     """
-    length = rec_segment.get_num_samples()
+    length = int(rec_segment.get_num_samples())
 
     if channel_indices is None:
         channel_indices = slice(None)

src/spikeinterface/core/segmentutils.py

@@ -156,7 +156,7 @@ def __init__(self, parent_segments, sampling_frequency, ignore_times=True):
         BaseRecordingSegment.__init__(self, **time_kwargs)
         self.parent_segments = parent_segments
         self.all_length = [rec_seg.get_num_samples() for rec_seg in self.parent_segments]
-        self.cumsum_length = np.cumsum([0] + self.all_length)
+        self.cumsum_length = [0] + [sum(self.all_length[: i + 1]) for i in range(len(self.all_length))]
         self.total_length = int(np.sum(self.all_length))
 
     def get_num_samples(self):

src/spikeinterface/postprocessing/amplitude_scalings.py

@@ -2,7 +2,7 @@
 
 import numpy as np
 
-from spikeinterface.core import ChannelSparsity, get_chunk_with_margin
+from spikeinterface.core import ChannelSparsity
 from spikeinterface.core.job_tools import ChunkRecordingExecutor, _shared_job_kwargs_doc, ensure_n_jobs, fix_job_kwargs
 
 from spikeinterface.core.template_tools import get_template_extremum_channel

src/spikeinterface/sortingcomponents/matching/naive.py

@@ -4,7 +4,7 @@
 
 
 import numpy as np
-from spikeinterface.core import get_noise_levels, get_channel_distances, get_chunk_with_margin, get_random_data_chunks
+from spikeinterface.core import get_noise_levels, get_channel_distances, get_random_data_chunks
 from spikeinterface.sortingcomponents.peak_detection import DetectPeakLocallyExclusive
 from spikeinterface.core.template import Templates
 

src/spikeinterface/sortingcomponents/peak_detection.py

@@ -24,7 +24,6 @@
 )
 
 from spikeinterface.postprocessing.unit_localization import get_convolution_weights
-from ..core import get_chunk_with_margin
 
 from .tools import make_multi_method_doc