From 114e098137c9384d596039b5319bbaf7efb010ce Mon Sep 17 00:00:00 2001 From: chrishalcrow <57948917+chrishalcrow@users.noreply.github.com> Date: Wed, 27 Mar 2024 08:19:38 +0000 Subject: [PATCH 1/2] Type bins for numba --- src/spikeinterface/postprocessing/isi.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/spikeinterface/postprocessing/isi.py b/src/spikeinterface/postprocessing/isi.py index 22aee972b9..c53c5542e4 100644 --- a/src/spikeinterface/postprocessing/isi.py +++ b/src/spikeinterface/postprocessing/isi.py @@ -137,7 +137,7 @@ def compute_isi_histograms_numba(sorting, window_ms: float = 50.0, bin_ms: float bin_size = int(round(fs * bin_ms * 1e-3)) window_size -= window_size % bin_size - bins = np.arange(0, window_size + bin_size, bin_size) # * 1e3 / fs + bins = np.arange(0, window_size + bin_size, bin_size, dtype=np.int64) # * 1e3 / fs spikes = sorting.to_spike_vector(concatenated=False) ISIs = np.zeros((num_units, len(bins) - 1), dtype=np.int64) From f6d2101fc79bc2946f857728da13d2a4bc101423 Mon Sep 17 00:00:00 2001 From: chrishalcrow <57948917+chrishalcrow@users.noreply.github.com> Date: Wed, 27 Mar 2024 08:20:00 +0000 Subject: [PATCH 2/2] Get rid of comment --- src/spikeinterface/postprocessing/isi.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/spikeinterface/postprocessing/isi.py b/src/spikeinterface/postprocessing/isi.py index c53c5542e4..3299164c07 100644 --- a/src/spikeinterface/postprocessing/isi.py +++ b/src/spikeinterface/postprocessing/isi.py @@ -137,7 +137,7 @@ def compute_isi_histograms_numba(sorting, window_ms: float = 50.0, bin_ms: float bin_size = int(round(fs * bin_ms * 1e-3)) window_size -= window_size % bin_size - bins = np.arange(0, window_size + bin_size, bin_size, dtype=np.int64) # * 1e3 / fs + bins = np.arange(0, window_size + bin_size, bin_size, dtype=np.int64) spikes = sorting.to_spike_vector(concatenated=False) ISIs = np.zeros((num_units, len(bins) - 1), dtype=np.int64)