Fix @mscheltienne review

vferat · Sep 12, 2023 · 7d616ce · 7d616ce
1 parent 04865be
commit 7d616ce
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 684 deletions.
diff --git a/draft.py b/draft.py
diff --git a/pycrostates/_typing.py b/pycrostates/_typing.py
@@ -29,7 +29,7 @@ class Cluster(ABC):
     pass
 
 
-class Segmentation(ABC):
+class Segmentation(_BaseSegmentation):
     """Typing for a clustering class."""
 
     pass

diff --git a/pycrostates/segmentation/tests/test_entropy.py b/pycrostates/segmentation/tests/test_entropy.py
@@ -1,6 +1,6 @@
 import numpy as np
 import pytest
-from mne import Epochs, make_fixed_length_events
+from mne import Epochs, make_fixed_length_epochs
 from mne.datasets import testing
 from mne.io import read_raw_fif
 
@@ -30,8 +30,7 @@
 raw = raw.pick("eeg").crop(0, 10)
 raw = raw.load_data().filter(1, 40).apply_proj()
 
-events = make_fixed_length_events(raw, 1)
-epochs = Epochs(raw, events, preload=True)
+epochs = make_fixed_length_epochs(raw, 1, preload=True)
 
 ModK_raw = ModKMeans(n_clusters=4, n_init=10, max_iter=100, tol=1e-4, random_state=1)
 ModK_epochs = ModKMeans(n_clusters=4, n_init=10, max_iter=100, tol=1e-4, random_state=1)