braindecode 1.8.1.dev170227516__tar.gz → 1.8.1.dev170439967__tar.gz

{braindecode-1.8.1.dev170227516 → braindecode-1.8.1.dev170439967}/NOTICE.txt

@@ -14,8 +14,7 @@ The following components are licensed under the Creative Commons Attribution-Non
 - `braindecode/models/eegminer.py`
 - `braindecode/models/meta_neuromotor.py`
 - `braindecode/models/brainmodule.py`
-
-As well as class later imported into the `braindecode.models.module` named as GeneralizedGaussianFilter.
+- `braindecode/models/neurorvq.py`
 
 The `meta_neuromotor.py` file is a derivative of
 `facebookresearch/generic-neuromotor-interface`, released by Meta Platforms,
@@ -43,6 +42,17 @@ The following components are licensed under the MIT License:
 - `braindecode/models/tcformer.py`
 - `braindecode/models/ifnet.py`
 - `braindecode/models/mirepnet.py`
+- `braindecode/models/popt.py`
+- `braindecode/modules/popt_modules.py`
+- `braindecode/models/brainomni.py`
+- `braindecode/modules/quantization.py`
+- `braindecode/models/seizure_transformer.py`
+- `docs/_static/model/seizure_transformer_arch.png`
+
+The `brainomni.py` file is a derivative of `OpenTSLab/BrainOmni` (MIT), whose
+SEANet codec derives from Meta's `facebookresearch/encodec` (MIT). The
+`quantization.py` module holds the residual vector quantization derived from
+`lucidrains/vector-quantize-pytorch` (MIT) and the BrainOmni port.
 
 ### Apache-2.0 Licensed Files
 
@@ -51,6 +61,7 @@ The following components are licensed under the Apache License 2.0:
 - `braindecode/models/mvpformer.py`
 - `braindecode/models/zuna.py`
 - `braindecode/models/luna.py`
+- `braindecode/models/mapa.py`
 - `braindecode/models/brant.py`
 - `braindecode/models/diver1.py`
 - `braindecode/models/mscformer.py`

{braindecode-1.8.1.dev170227516/braindecode.egg-info → braindecode-1.8.1.dev170439967}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: braindecode
-Version: 1.8.1.dev170227516
+Version: 1.8.1.dev170439967
 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>
@@ -44,6 +44,8 @@ Requires-Dist: huggingface_hub[torch]>=0.20.0; extra == "hub"
 Requires-Dist: zarr>=3.0; extra == "hub"
 Provides-Extra: viz
 Requires-Dist: captum>=0.7.0; extra == "viz"
+Provides-Extra: sae
+Requires-Dist: sae-lens==6.51.3; extra == "sae"
 Provides-Extra: tests
 Requires-Dist: pytest<9.1; extra == "tests"
 Requires-Dist: pytest-cov; extra == "tests"
@@ -58,6 +60,8 @@ Provides-Extra: typing
 Requires-Dist: exca>=0.5.10; extra == "typing"
 Requires-Dist: numpydantic>=1.7; extra == "typing"
 Provides-Extra: docs
+Requires-Dist: pytest<9.1; extra == "docs"
+Requires-Dist: sphinx-markdown-builder==0.6.11; extra == "docs"
 Requires-Dist: sphinx_gallery>=0.17; extra == "docs"
 Requires-Dist: sphinx_rtd_theme; extra == "docs"
 Requires-Dist: sphinx-autodoc-typehints; extra == "docs"

{braindecode-1.8.1.dev170227516 → braindecode-1.8.1.dev170439967}/braindecode/classifier.py

@@ -207,10 +207,11 @@ class EEGClassifier(_EEGNeuralNet, NeuralNetClassifier):
 
         Returns
         -------
-        trial_predictions : np.ndarray
+        trial_predictions : np.ndarray | list of np.ndarray
             3-dimensional array (n_trials x n_classes x n_predictions), where
             the number of predictions depend on the chosen window size and the
-            receptive field of the network.
+            receptive field of the network. If trials have different lengths,
+            a list with one (n_classes x n_predictions) array per trial.
         trial_targets : np.ndarray
             Ground-truth targets from the dataset in a 2-dimensional array
             (n_trials x n_targets). Only returned when ``return_targets=True``.

{braindecode-1.8.1.dev170227516 → braindecode-1.8.1.dev170439967}/braindecode/datasets/tuh.py

@@ -815,7 +815,8 @@ def _get_header(*args, **kwargs):
     for paths_dict in [_TUH_EEG_PATHS, _TUH_EEG_ABNORMAL_PATHS, _TUH_EEG_EVENTS_PATHS]:
         for version_dict in paths_dict.values():
             all_paths.update(version_dict)
-    return all_paths[args[0]]
+    # Mock keys use POSIX separators; pathlib emits backslashes on Windows.
+    return all_paths[os.fspath(args[0]).replace("\\", "/")]
 
 
 _TUH_EEG_PATHS = {

{braindecode-1.8.1.dev170227516 → braindecode-1.8.1.dev170439967}/braindecode/models/__init__.py

@@ -9,6 +9,7 @@ from .bendr import BENDR, InterpolatedBENDR
 from .biot import BIOT, InterpolatedBIOT
 from .brainbert import BrainBERT
 from .brainmodule import BrainModule
+from .brainomni import BrainOmni, BrainTokenizer
 from .brant import Brant
 from .cbramod import CBraMod
 from .codebrain import CodeBrain
@@ -40,6 +41,7 @@ from .ifnet import IFNet
 from .interpolated import InterpolatedModel
 from .labram import InterpolatedLaBraM, Labram
 from .luna import LUNA
+from .mapa import MAPA
 from .medformer import MEDFormer
 from .meta_neuromotor import MetaNeuromotorHand
 from .mirepnet import MIRepNet
@@ -47,9 +49,12 @@ from .mscformer import MSCFormer
 from .msvtnet import MSVTNet
 from .mvpformer import MVPFormer
 from .neuropose import NeuroPose
+from .neurorvq import NeuroRVQ
 from .patchedtransformer import PBT
+from .popt import PopulationTransformer
 from .reve import REVE
 from .sccnet import SCCNet
+from .seizure_transformer import SeizureTransformer
 from .sensingdynamics import SensingDynamics
 from .shallow_fbcsp import ShallowFBCSPNet
 from .signal_jepa import (
@@ -95,6 +100,8 @@ __all__ = [
     "BIOT",
     "BENDR",
     "BrainBERT",
+    "BrainOmni",
+    "BrainTokenizer",
     "CBraMod",
     "CodeBrain",
     "ContraWR",
@@ -138,14 +145,18 @@ __all__ = [
     "LUNA",
     "extract_channel_locations_from_chs_info",
     "positions_from_chs_info",
+    "MAPA",
     "MEDFormer",
     "MIRepNet",
+    "NeuroRVQ",
     "MSCFormer",
     "MSVTNet",
     "MVPFormer",
     "PBT",
+    "PopulationTransformer",
     "REVE",
     "SCCNet",
+    "SeizureTransformer",
     "ShallowFBCSPNet",
     "SignalJEPA",
     "SignalJEPA_Contextual",

{braindecode-1.8.1.dev170227516 → braindecode-1.8.1.dev170439967}/braindecode/models/base.py

@@ -779,6 +779,22 @@ class EEGModuleMixin(_BaseHubMixin, metaclass=_BraindecodeDocstringMeta):
 
     if HAS_HF_HUB:
 
+        @classmethod
+        def from_pretrained(cls, *args, **kwargs):
+            # The Hub mixin fills omitted __init__ args from config.json; pin the
+            # derived geometry arg so it cannot clash with the caller's one
+            # (e.g. caller chs_info vs saved n_chans) in __init__'s checks.
+            if "chs_info" in kwargs and "n_chans" not in kwargs:
+                chs_info = kwargs["chs_info"]
+                kwargs["n_chans"] = None if chs_info is None else len(chs_info)
+            elif "n_chans" in kwargs and "chs_info" not in kwargs:
+                kwargs["chs_info"] = None
+            if ("n_times" in kwargs or "sfreq" in kwargs) and (
+                "input_window_seconds" not in kwargs
+            ):
+                kwargs["input_window_seconds"] = None
+            return super().from_pretrained(*args, **kwargs)
+
         @classmethod
         def _from_pretrained(
             cls,