braindecode 1.8.1.dev174857482__tar.gz → 1.8.1.dev174954496__tar.gz

{braindecode-1.8.1.dev174857482 → braindecode-1.8.1.dev174954496}/NOTICE.txt

@@ -45,6 +45,15 @@ The following components are licensed under the MIT License:
 - `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
 
@@ -53,6 +62,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.dev174857482/braindecode.egg-info → braindecode-1.8.1.dev174954496}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: braindecode
-Version: 1.8.1.dev174857482
+Version: 1.8.1.dev174954496
 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>

{braindecode-1.8.1.dev174857482 → braindecode-1.8.1.dev174954496}/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
@@ -51,6 +53,7 @@ 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 (
@@ -96,6 +99,8 @@ __all__ = [
     "BIOT",
     "BENDR",
     "BrainBERT",
+    "BrainOmni",
+    "BrainTokenizer",
     "CBraMod",
     "CodeBrain",
     "ContraWR",
@@ -139,6 +144,7 @@ __all__ = [
     "LUNA",
     "extract_channel_locations_from_chs_info",
     "positions_from_chs_info",
+    "MAPA",
     "MEDFormer",
     "MIRepNet",
     "MSCFormer",
@@ -148,6 +154,7 @@ __all__ = [
     "PopulationTransformer",
     "REVE",
     "SCCNet",
+    "SeizureTransformer",
     "ShallowFBCSPNet",
     "SignalJEPA",
     "SignalJEPA_Contextual",