preload Schirrmsister2017 - Fixes NeuroTechX#289

PierreGtch · May 9, 2022 · 333925c · 333925c
1 parent d4305c4
commit 333925c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/moabb/datasets/schirrmeister2017.py b/moabb/datasets/schirrmeister2017.py
@@ -80,9 +80,9 @@ def _url(prefix):
             for t in ["train", "test"]
         ]
 
-    def _get_single_subject_data(self, subject):
+    def _get_single_subject_data(self, subject, preload=True):
         train_raw, test_raw = [
-            mne.io.read_raw_edf(path, infer_types=True)
+            mne.io.read_raw_edf(path, infer_types=True, preload=preload)
             for path in self.data_path(subject)
         ]