Braindecode 1.3.0.dev183374645__tar.gz → 1.3.0.dev183770261__tar.gz

{braindecode-1.3.0.dev183374645/braindecode.egg-info → braindecode-1.3.0.dev183770261}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: braindecode
-Version: 1.3.0.dev183374645
+Version: 1.3.0.dev183770261
 Summary: Deep learning software to decode EEG, ECG or MEG signals
 Author-email: Robin Tibor Schirrmeister <robintibor@gmail.com>, Bruno Aristimunha Pinto <b.aristimunha@gmail.com>, Alexandre Gramfort <agramfort@meta.com>
 Maintainer-email: Alexandre Gramfort <agramfort@meta.com>, Bruno Aristimunha Pinto <b.aristimunha@gmail.com>, Robin Tibor Schirrmeister <robintibor@gmail.com>
@@ -22,12 +22,17 @@ Requires-Python: >=3.11
 Description-Content-Type: text/x-rst
 License-File: LICENSE.txt
 License-File: NOTICE.txt
-Requires-Dist: torch>=2.2
-Requires-Dist: torchaudio>=2.0
 Requires-Dist: mne>=1.11.0
 Requires-Dist: mne_bids>=0.16
+Requires-Dist: numpy
+Requires-Dist: pandas
+Requires-Dist: scipy
+Requires-Dist: matplotlib
 Requires-Dist: h5py
 Requires-Dist: skorch>=1.3.0
+Requires-Dist: torch>=2.0
+Requires-Dist: torchaudio>=2.0
+Requires-Dist: einops
 Requires-Dist: joblib
 Requires-Dist: torchinfo
 Requires-Dist: wfdb
@@ -47,8 +52,6 @@ Requires-Dist: pytest-cov; extra == "tests"
 Requires-Dist: codecov; extra == "tests"
 Requires-Dist: pytest_cases; extra == "tests"
 Requires-Dist: mypy; extra == "tests"
-Requires-Dist: transformers>=4.57.0; extra == "tests"
-Requires-Dist: bids_validator; extra == "tests"
 Provides-Extra: typing
 Requires-Dist: exca==0.4; extra == "typing"
 Requires-Dist: numpydantic>=1.7; extra == "typing"
@@ -72,7 +75,6 @@ Requires-Dist: openneuro-py; extra == "docs"
 Requires-Dist: plotly; extra == "docs"
 Requires-Dist: shap; extra == "docs"
 Requires-Dist: nbformat; extra == "docs"
-Requires-Dist: transformers; extra == "docs"
 Provides-Extra: all
 Requires-Dist: braindecode[moabb]; extra == "all"
 Requires-Dist: braindecode[tests]; extra == "all"

{braindecode-1.3.0.dev183374645 → braindecode-1.3.0.dev183770261}/braindecode/datasets/base.py

@@ -28,7 +28,7 @@ from mne.utils.docs import deprecated
 from torch.utils.data import ConcatDataset, Dataset
 from typing_extensions import TypeVar
 
-from .bids.hub import HubDatasetMixin
+from .hub import HubDatasetMixin
 from .registry import register_dataset
 
 
@@ -719,14 +719,8 @@ class BaseConcatDataset(ConcatDataset, HubDatasetMixin, Generic[T]):
             hasattr(self.datasets[0], "raw") or hasattr(self.datasets[0], "windows")
         ):
             raise ValueError("dataset should have either raw or windows attribute")
-
-        # Create path if it doesn't exist
-        os.makedirs(path, exist_ok=True)
-
         path_contents = os.listdir(path)
-        n_sub_dirs = len(
-            [e for e in path_contents if os.path.isdir(os.path.join(path, e))]
-        )
+        n_sub_dirs = len([os.path.isdir(e) for e in path_contents])
         for i_ds, ds in enumerate(self.datasets):
             # remove subdirectory from list of untouched files / subdirectories
             if str(i_ds + offset) in path_contents:

braindecode-1.3.0.dev183374645/braindecode/datasets/bids/datasets.py → braindecode-1.3.0.dev183770261/braindecode/datasets/bids.py

@@ -1,4 +1,3 @@
-# mypy: ignore-errors
 """Dataset for loading BIDS.
 
 More information on BIDS (Brain Imaging Data Structure) can be found at https://bids.neuroimaging.io
@@ -20,19 +19,26 @@ import numpy as np
 import pandas as pd
 from joblib import Parallel, delayed
 
-from ..base import BaseConcatDataset, RawDataset, WindowsDataset
+from .base import BaseConcatDataset, RawDataset, WindowsDataset
 
 
 def _description_from_bids_path(bids_path: mne_bids.BIDSPath) -> dict[str, Any]:
-    description = {"path": bids_path.fpath, **bids_path.entities}
-    description.update(
-        {
-            "suffix": bids_path.suffix,
-            "extension": bids_path.extension,
-            "datatype": bids_path.datatype,
-        }
-    )
-    return description
+    return {
+        "path": bids_path.fpath,
+        "subject": bids_path.subject,
+        "session": bids_path.session,
+        "task": bids_path.task,
+        "acquisition": bids_path.acquisition,
+        "run": bids_path.run,
+        "processing": bids_path.processing,
+        "recording": bids_path.recording,
+        "space": bids_path.space,
+        "split": bids_path.split,
+        "description": bids_path.description,
+        "suffix": bids_path.suffix,
+        "extension": bids_path.extension,
+        "datatype": bids_path.datatype,
+    }
 
 
 @dataclass

braindecode-1.3.0.dev183374645/braindecode/datasets/bids/iterable.py → braindecode-1.3.0.dev183770261/braindecode/datasets/experimental.py

@@ -25,13 +25,14 @@ class BIDSIterableDataset(IterableDataset):
 
     Examples
     --------
-    >>> from braindecode.datasets import BaseConcatDataset, RawDataset, RecordDataset
-    >>> from braindecode.datasets.bids import BIDSIterableDataset
+    >>> from braindecode.datasets import RecordDataset, BaseConcatDataset
+    >>> from braindecode.datasets.bids import BIDSIterableDataset, _description_from_bids_path
     >>> from braindecode.preprocessing import create_fixed_length_windows
     >>>
     >>> def my_reader_fn(path):
     ...     raw = mne_bids.read_raw_bids(path)
-    ...     ds: RecordDataset = RawDataset(raw, description={"path": path.fpath})
+    ...     desc = _description_from_bids_path(path)
+    ...     ds = RawDataset(raw, description=desc)
     ...     windows_ds = create_fixed_length_windows(
     ...         BaseConcatDataset([ds]),
     ...         window_size_samples=400,
@@ -47,8 +48,7 @@ class BIDSIterableDataset(IterableDataset):
     Parameters
     ----------
     reader_fn : Callable[[mne_bids.BIDSPath], Sequence]
-        A function that takes a BIDSPath and returns a dataset (e.g., a
-        RecordDataset or BaseConcatDataset of RecordDataset).
+        A function that takes a BIDSPath and returns a dataset.
     pool_size : int
         The number of recordings to read and sample from.
     bids_paths : list[mne_bids.BIDSPath] | None