Merge remote-tracking branch 'origin/eval_splitters' into eval_splitters

NeuroTechX · Oct 1, 2024 · b29ecd2 · b29ecd2
2 parents 516a5e8 + e5159f2
commit b29ecd2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 6 deletions.
diff --git a/moabb/evaluations/metasplitters.py b/moabb/evaluations/metasplitters.py
@@ -91,7 +91,11 @@ def split(self, X, y, metadata):
 
             for session in sessions:
                 session_mask = meta_["session"] == session
-                _, _, meta_session = X_[session_mask], y_[session_mask], meta_[session_mask]
+                _, _, meta_session = (
+                    X_[session_mask],
+                    y_[session_mask],
+                    meta_[session_mask],
+                )
 
                 # If you can (amd want) to split by run also
                 if self.run and "run" in meta_session.columns:

diff --git a/moabb/tests/metasplits.py b/moabb/tests/metasplits.py
@@ -4,16 +4,14 @@
 
 from moabb.datasets.fake import FakeDataset
 from moabb.evaluations.metasplitters import OfflineSplit, PseudoOnlineSplit, SamplerSplit
-from moabb.evaluations.splitters import (
-    CrossSessionSplitter,
-    CrossSubjectSplitter,
-)
+from moabb.evaluations.splitters import CrossSessionSplitter, CrossSubjectSplitter
 from moabb.paradigms.motor_imagery import FakeImageryParadigm
 
 
 dataset = FakeDataset(["left_hand", "right_hand"], n_subjects=3, seed=12)
 paradigm = FakeImageryParadigm()
 
+
 # Still working on this
 def eval_sampler_split():
     for subject in dataset.subject_list:
@@ -151,4 +149,3 @@ def test_sampler(data_eval):
         # Check if the output is the same as the input
         assert np.array_equal(X_train, X_train_t)
         assert np.array_equal(X_test, X_test_t)
-