Braindecode 1.3.0.dev176122045__tar.gz → 1.3.0.dev176529508__tar.gz

{braindecode-1.3.0.dev176122045/braindecode.egg-info → braindecode-1.3.0.dev176529508}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: braindecode
-Version: 1.3.0.dev176122045
+Version: 1.3.0.dev176529508
 Summary: Deep learning software to decode EEG, ECG or MEG signals
 Author-email: Robin Tibor Schirrmeister <robintibor@gmail.com>
 Maintainer-email: Alexandre Gramfort <agramfort@meta.com>, Bruno Aristimunha Pinto <b.aristimunha@gmail.com>, Robin Tibor Schirrmeister <robintibor@gmail.com>
@@ -38,6 +38,7 @@ Requires-Dist: wfdb
 Requires-Dist: h5py
 Requires-Dist: linear_attention_transformer
 Requires-Dist: docstring_inheritance
+Requires-Dist: rotary_embedding_torch
 Provides-Extra: moabb
 Requires-Dist: moabb>=1.2.0; extra == "moabb"
 Provides-Extra: eegprep

{braindecode-1.3.0.dev176122045 → braindecode-1.3.0.dev176529508}/braindecode/datasets/base.py

@@ -19,7 +19,7 @@ import warnings
 from abc import abstractmethod
 from collections.abc import Callable
 from glob import glob
-from typing import Generic, Iterable, no_type_check
+from typing import Any, Generic, Iterable, no_type_check
 
 import mne.io
 import numpy as np
@@ -129,6 +129,7 @@ class RawDataset(RecordDataset):
 
         # save target name for load/save later
         self.target_name = self._target_name(target_name)
+        self.raw_preproc_kwargs: list[dict[str, Any]] = []
 
     def __getitem__(self, index):
         X = self.raw[:, index][0]
@@ -235,6 +236,7 @@ class EEGWindowsDataset(RecordDataset):
         ].to_numpy()
         if self.targets_from == "metadata":
             self.y = metadata.loc[:, "target"].to_list()
+        self.raw_preproc_kwargs: list[dict[str, Any]] = []
 
     def __getitem__(self, index: int):
         """Get a window and its target.
@@ -334,6 +336,8 @@ class WindowsDataset(RecordDataset):
         ].to_numpy()
         if self.targets_from == "metadata":
             self.y = metadata.loc[:, "target"].to_list()
+        self.raw_preproc_kwargs: list[dict[str, Any]] = []
+        self.window_preproc_kwargs: list[dict[str, Any]] = []
 
     def __getitem__(self, index: int):
         """Get a window and its target.
@@ -794,7 +798,7 @@ class BaseConcatDataset(ConcatDataset, Generic[T]):
                 kwargs = getattr(ds, kwargs_name)
                 if kwargs is not None:
                     with open(kwargs_file_path, "w") as f:
-                        json.dump(kwargs, f)
+                        json.dump(kwargs, f, indent=2)
 
     @staticmethod
     def _save_target_name(sub_dir, ds):

{braindecode-1.3.0.dev176122045 → braindecode-1.3.0.dev176529508}/braindecode/datautil/serialization.py

@@ -302,7 +302,6 @@ def _load_kwargs_json(kwargs_name, sub_dir):
     kwargs_file_path = os.path.join(sub_dir, kwargs_file_name)
     if os.path.exists(kwargs_file_path):
         kwargs = json.load(open(kwargs_file_path, "r"))
-        kwargs = [tuple(kwarg) for kwarg in kwargs]
         return kwargs
 
 

{braindecode-1.3.0.dev176122045 → braindecode-1.3.0.dev176529508}/braindecode/models/__init__.py

@@ -28,6 +28,8 @@ from .fbmsnet import FBMSNet
 from .hybrid import HybridNet
 from .ifnet import IFNet
 from .labram import Labram
+from .luna import LUNA
+from .medformer import MEDFormer
 from .msvtnet import MSVTNet
 from .patchedtransformer import PBT
 from .sccnet import SCCNet
@@ -48,7 +50,11 @@ from .tcn import BDTCN, TCN
 from .tidnet import TIDNet
 from .tsinception import TSception
 from .usleep import USleep
-from .util import _init_models_dict, models_mandatory_parameters
+from .util import (
+    _init_models_dict,
+    extract_channel_locations_from_chs_info,
+    models_mandatory_parameters,
+)
 
 # Call this last in order to make sure the dataset list is populated with
 # the models imported in this file.
@@ -82,6 +88,9 @@ __all__ = [
     "HybridNet",
     "IFNet",
     "Labram",
+    "LUNA",
+    "extract_channel_locations_from_chs_info",
+    "MEDFormer",
     "MSVTNet",
     "PBT",
     "SCCNet",