Merge branch 'dev-meeg' of https://github.com/vferat/pycrostates into…

… dev-meeg
vferat · Feb 20, 2024 · c86ab92 · c86ab92
2 parents f9ba5fc + 9387250
commit c86ab92
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/pycrostates/cluster/_base.py b/pycrostates/cluster/_base.py
@@ -953,7 +953,9 @@ def _segment(
         half_window_size: int,
     ) -> NDArray[int]:
         """Create segmentation. Must operate on a copy of states."""
-        corr = _correlation(states.T, data, ignore_polarity=ignore_polarity)[:len(states), :]
+        corr = _correlation(states.T, data, ignore_polarity=ignore_polarity)[
+            : len(states), :
+        ]
         if ignore_polarity:
             corr = np.abs(corr)
         labels = np.argmax(corr, axis=0)