Numba code more readable

SpikeInterface · Nov 20, 2023 · 8950077 · 8950077
1 parent 3f71517
commit 8950077
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/spikeinterface/core/sorting_tools.py b/src/spikeinterface/core/sorting_tools.py
@@ -79,8 +79,9 @@ def vector_to_list_of_spiketrain_numba(sample_indices, unit_indices, num_units):
 
         current_x = np.zeros(num_units, dtype=np.int64)
         for s in range(num_spikes):
-            spike_trains[unit_indices[s]][current_x[unit_indices[s]]] = sample_indices[s]
-            current_x[unit_indices[s]] += 1
+            unit_index = unit_indices[s]
+            spike_trains[unit_index][current_x[unit_index]] = sample_indices[s]
+            current_x[unit_index] += 1
 
         return spike_trains