diff --git a/src/spikeinterface/sortingcomponents/peak_detection.py b/src/spikeinterface/sortingcomponents/peak_detection.py index 5751ba2526..0fe1764579 100644 --- a/src/spikeinterface/sortingcomponents/peak_detection.py +++ b/src/spikeinterface/sortingcomponents/peak_detection.py @@ -650,7 +650,7 @@ def __init__( self.num_channels = recording.get_num_channels() self.num_templates = self.num_channels - if peak_sign == 'both': + if peak_sign == "both": weights = np.repeat(weights, 2, axis=1) weights[1::2] *= -1 self.num_templates *= 2 @@ -717,7 +717,7 @@ def compute(self, traces, start_frame, end_frame, segment_index, max_margin): # Find peaks and correct for time shift z_ind, peak_chan_ind, peak_sample_ind = np.nonzero(peak_mask) - if self.peak_sign == 'both': + if self.peak_sign == "both": peak_chan_ind = peak_chan_ind % self.num_channels # If we want to estimate z