diff --git a/src/spikeinterface/core/node_pipeline.py b/src/spikeinterface/core/node_pipeline.py index 9b61ec0dab..a6dabf77b5 100644 --- 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 index 6f8a8aabcb..0e471444d8 100644 --- 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", ),