oups

SpikeInterface · Oct 25, 2023 · 6887c97 · 6887c97
1 parent 0c790f4
commit 6887c97
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/spikeinterface/core/node_pipeline.py b/src/spikeinterface/core/node_pipeline.py
@@ -215,7 +215,7 @@ def compute(self, traces, start_frame, end_frame, segment_index, max_margin):
 
         # handle amplitude
         for i, peak in enumerate(local_peaks):
-            local_peaks["amplitude"][i] = traces[local_peaks["sample_index"], local_peaks[i]["channel_index"]]
+            local_peaks["amplitude"][i] = traces[peak["sample_index"], peak["channel_index"]]
 
         return (local_peaks,)
 

diff --git a/src/spikeinterface/postprocessing/spike_locations.py b/src/spikeinterface/postprocessing/spike_locations.py
@@ -31,7 +31,7 @@ def _set_params(
         ms_before=0.5,
         ms_after=0.5,
         spike_retriver_kwargs=dict(
-            channel_from_template=False,
+            channel_from_template=True,
             radius_um=50,
             peaks_sign="neg",
         ),
@@ -125,7 +125,7 @@ def compute_spike_locations(
     ms_before=0.5,
     ms_after=0.5,
     spike_retriver_kwargs=dict(
-        channel_from_template=False,
+        channel_from_template=True,
         radius_um=50,
         peaks_sign="neg",
     ),