[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
SpikeInterface · Oct 11, 2024 · 0102b86 · 0102b86
1 parent bdbead2
commit 0102b86
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/spikeinterface/sortingcomponents/matching/kilosort.py b/src/spikeinterface/sortingcomponents/matching/kilosort.py
@@ -132,7 +132,7 @@ def compute_matching(self, traces, start_frame, end_frame, segment_index):
             for t in range(self.max_iter):
                 Cf = torch.relu(B) ** 2 / self.nm.unsqueeze(-1)
                 Cf[:, : self.margin] = 0
-                Cf[:, -self.margin:] = 0
+                Cf[:, -self.margin :] = 0
 
                 Cfmax, imax = torch.max(Cf, 0)
                 Cmax = max_pool1d(
@@ -162,7 +162,7 @@ def compute_matching(self, traces, start_frame, end_frame, segment_index):
 
             spikes = spikes[:k]
             spikes["channel_index"] = 0
-            spikes["sample_index"] += (self.nbefore-self.nafter)//2
+            spikes["sample_index"] += (self.nbefore - self.nafter) // 2
             order = np.argsort(spikes["sample_index"])
             spikes = spikes[order]