diff --git a/pycrostates/cluster/_base.py b/pycrostates/cluster/_base.py index 679075be..a5080654 100644 --- a/pycrostates/cluster/_base.py +++ b/pycrostates/cluster/_base.py @@ -1059,9 +1059,10 @@ def _reject_short_segments( ignore_polarity=ignore_polarity, )[0] right_corr = _correlation( - data[:, right], - data[:, right + 1], - ignore_polarity=ignore_polarity)[0] + data[:, right], + data[:, right + 1], + ignore_polarity=ignore_polarity, + )[0] if np.abs(right_corr - left_corr) <= 1e-8: # equal corr, try to do both sides if len(new_segment) == 1: