Patch for neuroconv (#2245)

* sorting tools imports * patch for neuroconv
SpikeInterface · Nov 23, 2023 · 4b10e37 · 4b10e37
1 parent 48a543e
commit 4b10e37
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/spikeinterface/core/sorting_tools.py b/src/spikeinterface/core/sorting_tools.py
@@ -1,7 +1,8 @@
+from __future__ import annotations
 import numpy as np
 
 
-def spike_vector_to_spike_trains(spike_vector: list[np.array], unit_ids: np.array) -> dict[dict]:
+def spike_vector_to_spike_trains(spike_vector: list[np.array], unit_ids: np.array) -> dict[dict[str, np.array]]:
     """
     Computes all spike trains for all units/segments from a spike vector list.