From 50a562ee6cd9d467ffb2304616bd1d2419ad1a2e Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Wed, 29 May 2024 05:56:41 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- src/spikeinterface/sortingcomponents/peak_detection.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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