Braindecode 1.8.0.dev174412347__tar.gz → 1.8.0.dev176127673__tar.gz

braindecode-1.8.0.dev176127673/NOTICE.txt

@@ -0,0 +1,25 @@
+# BRAINDECODE Notice
+
+## Licensed Components
+
+### BSD-3-Clause Licensed Files
+
+All files within the `src/` directory are licensed under the BSD-3-Clause License.
+
+### CC BY-NC 4.0 Licensed Files
+
+The following components are licensed under the Creative Commons Attribution-NonCommercial 4.0 International License:
+
+- `braindecode/models/eegminer.py`
+- `braindecode/models/meta_neuromotor.py`
+
+As well as class later imported into the `braindecode.models.module` named as GeneralizedGaussianFilter.
+
+The `meta_neuromotor.py` file is a derivative of
+`facebookresearch/generic-neuromotor-interface`, released by Meta Platforms,
+Inc. under CC BY-NC 4.0, and inherits the same noncommercial terms.
+
+## License Links
+
+- [BSD-3-Clause License](https://opensource.org/licenses/BSD-3-Clause)
+- [CC BY-NC 4.0 License](https://creativecommons.org/licenses/by-nc/4.0/)

{braindecode-1.8.0.dev174412347/braindecode.egg-info → braindecode-1.8.0.dev176127673}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: braindecode
-Version: 1.8.0.dev174412347
+Version: 1.8.0.dev176127673
 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>
@@ -50,7 +50,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: ipython; extra == "tests"
 Requires-Dist: transformers>=4.57.0; extra == "tests"
 Requires-Dist: bids_validator; extra == "tests"
 Provides-Extra: typing
@@ -260,10 +259,7 @@ This project is primarily licensed under the BSD-3-Clause License.
 Additional Components
 =====================
 
-Some components within this repository are licensed under other licenses, including
-Creative Commons Attribution-NonCommercial 4.0 International (CC BY-NC 4.0), Creative
-Commons Attribution-NonCommercial-ShareAlike 4.0 International (CC BY-NC-SA 4.0), MIT
-and Apache-2.0.
+Some components within this repository are licensed under the Creative Commons
+Attribution-NonCommercial 4.0 International License.
 
-Please refer to the ``LICENSE`` and ``NOTICE`` files for the per-file list and more
-detailed information.
+Please refer to the ``LICENSE`` and ``NOTICE`` files for more detailed information.

{braindecode-1.8.0.dev174412347 → braindecode-1.8.0.dev176127673}/README.rst

@@ -170,10 +170,7 @@ This project is primarily licensed under the BSD-3-Clause License.
 Additional Components
 =====================
 
-Some components within this repository are licensed under other licenses, including
-Creative Commons Attribution-NonCommercial 4.0 International (CC BY-NC 4.0), Creative
-Commons Attribution-NonCommercial-ShareAlike 4.0 International (CC BY-NC-SA 4.0), MIT
-and Apache-2.0.
+Some components within this repository are licensed under the Creative Commons
+Attribution-NonCommercial 4.0 International License.
 
-Please refer to the ``LICENSE`` and ``NOTICE`` files for the per-file list and more
-detailed information.
+Please refer to the ``LICENSE`` and ``NOTICE`` files for more detailed information.

{braindecode-1.8.0.dev174412347 → braindecode-1.8.0.dev176127673}/braindecode/datasets/base.py

@@ -32,7 +32,6 @@ from mne.utils.docs import deprecated
 from torch.utils.data import ConcatDataset, Dataset, IterableDataset
 from typing_extensions import TypeVar
 
-from ._notebook_viewer import plot as _viewer_plot
 from .bids.hub import HubDatasetMixin
 from .bids.hub_io import _restore_nan_from_json
 from .registry import register_dataset
@@ -1175,8 +1174,6 @@ class BaseConcatDataset(ConcatDataset, HubDatasetMixin, Generic[T]):
         If True, defer computing cumulative sizes until length or item access.
     """
 
-    plot = _viewer_plot  # eegdash-viewer embed; a direct member so the API docs list it
-
     datasets: list[T]
 
     def __init__(

{braindecode-1.8.0.dev174412347 → braindecode-1.8.0.dev176127673}/braindecode/functional/functions.py

@@ -40,7 +40,7 @@ def drop_path(
     x : torch.Tensor
         input tensor
     drop_prob : float, optional
-        probability of dropping a path, by default 0.0
+        survival rate (i.e. probability of being kept), by default 0.0
     training : bool, optional
         whether the model is in training mode, by default False
     scale_by_keep : bool, optional
@@ -68,10 +68,9 @@ def drop_path(
     shape = (x.shape[0],) + (1,) * (
         x.ndim - 1
     )  # work with diff dim tensors, not just 2D ConvNets
-    probabilities = torch.full(shape, keep_prob, dtype=torch.float32, device=x.device)
-    random_tensor = torch.bernoulli(probabilities).to(dtype=x.dtype)
+    random_tensor = x.new_empty(shape).bernoulli_(keep_prob)
     if keep_prob > 0.0 and scale_by_keep:
-        random_tensor = random_tensor / keep_prob
+        random_tensor.div_(keep_prob)
     return x * random_tensor
 
 

{braindecode-1.8.0.dev174412347 → braindecode-1.8.0.dev176127673}/braindecode/models/__init__.py

@@ -40,11 +40,9 @@ from .medformer import MEDFormer
 from .meta_neuromotor import MetaNeuromotorHand
 from .msvtnet import MSVTNet
 from .mvpformer import MVPFormer
-from .neuropose import NeuroPose
 from .patchedtransformer import PBT
 from .reve import REVE
 from .sccnet import SCCNet
-from .sensingdynamics import SensingDynamics
 from .shallow_fbcsp import ShallowFBCSPNet
 from .signal_jepa import (
     InterpolatedSignalJEPA,
@@ -73,7 +71,6 @@ from .util import (
     models_mandatory_parameters,
     positions_from_chs_info,
 )
-from .vemg2pose import VEMG2Pose
 from .zuna import ZUNA
 
 # Call this last in order to make sure the dataset list is populated with
@@ -110,9 +107,6 @@ __all__ = [
     "EEGTCNet",
     "DGCNN",
     "EMG2QwertyNet",
-    "NeuroPose",
-    "SensingDynamics",
-    "VEMG2Pose",
     "FBCNet",
     "FBLightConvNet",
     "FBMSNet",

{braindecode-1.8.0.dev174412347 → braindecode-1.8.0.dev176127673}/braindecode/models/base.py

@@ -519,31 +519,6 @@ class EEGModuleMixin(_BaseHubMixin, metaclass=_BraindecodeDocstringMeta):
         resolve_type_kwargs(cls, config)
         return cls(**config)
 
-    def _set_n_outputs(self, n_outputs: int) -> None:
-        """Record a new ``n_outputs`` on the model and its saved configs.
-
-        Helper for :meth:`reset_head` implementations: validates the value,
-        then keeps the ``n_outputs`` property, the braindecode init kwargs
-        and the Hugging Face hub config in sync, so a re-serialized model
-        reports the head it actually has.
-
-        Parameters
-        ----------
-        n_outputs : int
-            New number of outputs. Must be positive.
-
-        .. versionadded:: 1.8
-        """
-        if n_outputs <= 0:
-            raise ValueError(f"n_outputs must be positive; got {n_outputs}.")
-        self._n_outputs = n_outputs
-        for config in (
-            getattr(self, "_braindecode_init_kwargs", None),
-            getattr(self, "_hub_mixin_config", None),
-        ):
-            if config is not None and "n_outputs" in config:
-                config["n_outputs"] = n_outputs
-
     def reset_head(self, n_outputs):
         """Replace the classification head for a new number of outputs.
 

{braindecode-1.8.0.dev174412347 → braindecode-1.8.0.dev176127673}/braindecode/models/brainmodule.py

@@ -25,7 +25,7 @@ from braindecode.modules.layers import SubjectLayers
 __all__ = ["BrainModule"]
 
 
-class BrainModule(EEGModuleMixin, nn.Module, license="cc-by-nc-4.0"):
+class BrainModule(EEGModuleMixin, nn.Module):
     r"""BrainModule from [brainmagick]_, also known as SimpleConv.
 
     A dilated convolutional encoder for EEG decoding, using residual

{braindecode-1.8.0.dev174412347 → braindecode-1.8.0.dev176127673}/braindecode/models/ctnet.py

@@ -24,7 +24,7 @@ from braindecode.modules import (
 )
 
 
-class CTNet(EEGModuleMixin, nn.Module, license="mit"):
+class CTNet(EEGModuleMixin, nn.Module):
     r"""CTNet from Zhao, W et al (2024) [ctnet]_.
 
     :bdg-success:`Convolution` :bdg-info:`Attention/Transformer`

{braindecode-1.8.0.dev174412347 → braindecode-1.8.0.dev176127673}/braindecode/models/dance.py

@@ -22,7 +22,7 @@ from braindecode.modules import ChannelMerger, Perceiver, SimpleConv
 from braindecode.modules.dance_modules import DanceDetrDecoder
 
 
-class DANCE(EEGModuleMixin, nn.Module, license="mit"):
+class DANCE(EEGModuleMixin, nn.Module):
     r"""DANCE from Lévy et al (2026) [dance]_.
 
     :bdg-success:`Convolution` :bdg-info:`Attention/Transformer` :bdg-dark-line:`Channel`
@@ -349,7 +349,8 @@ class DANCE(EEGModuleMixin, nn.Module, license="mit"):
 
     def reset_head(self, n_outputs: int) -> None:
         """Replace the dense head and the DETR class head for a new ``n_outputs``."""
-        self._set_n_outputs(n_outputs)
+        if n_outputs <= 0:
+            raise ValueError(f"n_outputs must be positive; got {n_outputs}.")
         old = self.final_layer
         self.final_layer = nn.Linear(old.in_features, n_outputs).to(
             device=old.weight.device, dtype=old.weight.dtype
@@ -358,3 +359,10 @@ class DANCE(EEGModuleMixin, nn.Module, license="mit"):
         self.decoder.class_head = nn.Linear(ch.in_features, n_outputs).to(
             device=ch.weight.device, dtype=ch.weight.dtype
         )
+        self._n_outputs = n_outputs
+        init_kwargs = getattr(self, "_braindecode_init_kwargs", None)
+        if init_kwargs is not None and "n_outputs" in init_kwargs:
+            init_kwargs["n_outputs"] = n_outputs
+        hub_config = getattr(self, "_hub_mixin_config", None)
+        if hub_config is not None and "n_outputs" in hub_config:
+            hub_config["n_outputs"] = n_outputs

{braindecode-1.8.0.dev174412347 → braindecode-1.8.0.dev176127673}/braindecode/models/eegconformer.py

@@ -185,7 +185,7 @@ class EEGConformer(EEGModuleMixin, nn.Module):
        31, pp.710-719. https://ieeexplore.ieee.org/document/9991178
     .. [ConformerCode] Song, Y., Zheng, Q., Liu, B. and Gao, X., 2022. EEG
        conformer: Convolutional transformer for EEG decoding and visualization.
-       https://github.com/eeyhsong/EEG-Conformer
+       https://github.com/eeyhsong/EEG-Conformer.
     """
 
     def __init__(

{braindecode-1.8.0.dev174412347 → braindecode-1.8.0.dev176127673}/braindecode/models/eegminer.py

@@ -6,7 +6,6 @@
 """
 
 # Authors: Sarthak Tayal <sarthaktayal2@gmail.com>
-# License: Creative Commons Attribution-NonCommercial 4.0 International (CC BY-NC 4.0)
 
 import torch
 from torch import nn
@@ -49,7 +48,7 @@ class _EEGMinerFeatures(nn.Module):
         return x[:, self.triu_row, self.triu_col, :]
 
 
-class EEGMiner(EEGModuleMixin, nn.Module, license="cc-by-nc-4.0"):
+class EEGMiner(EEGModuleMixin, nn.Module):
     r"""EEGMiner from Ludwig et al (2024) [eegminer]_.
 
     :bdg-success:`Convolution` :bdg-warning:`Interpretability`

{braindecode-1.8.0.dev174412347 → braindecode-1.8.0.dev176127673}/braindecode/models/eegsimpleconv.py

@@ -93,7 +93,7 @@ class EEGSimpleConv(EEGModuleMixin, torch.nn.Module):
     .. [Yassine2023Code] Yassine El Ouahidi, V. Gripon, B. Pasdeloup, G. Bouallegue
         N. Farrugia, G. Lioi, 2023. A Strong and Simple Deep Learning Baseline for
         BCI Motor Imagery Decoding. GitHub repository.
-        https://github.com/elouayas/EEGSimpleConv
+        https://github.com/elouayas/EEGSimpleConv.
 
     """
 

{braindecode-1.8.0.dev174412347 → braindecode-1.8.0.dev176127673}/braindecode/models/emg2qwerty.py

@@ -1,10 +1,9 @@
 # Authors: Meta Platforms, Inc. and affiliates (original emg2qwerty)
 #          Bruno Aristimunha <b.aristimunha@gmail.com> (Braindecode adaptation)
 #
-# License: Creative Commons Attribution-NonCommercial-ShareAlike 4.0
-#          International (CC BY-NC-SA 4.0)
+# License: Creative Commons Attribution-NonCommercial 4.0 International (CC BY-NC 4.0)
 # Adapted from https://github.com/facebookresearch/emg2qwerty (NeurIPS 2024).
-# Inherits CC BY-NC-SA 4.0; not covered by braindecode's BSD-3 license.
+# Inherits CC BY-NC 4.0; not covered by braindecode's BSD-3 license.
 """``EMG2QwertyNet``: TDS-Conv-CTC for sEMG-to-keystroke decoding."""
 
 from __future__ import annotations
@@ -17,10 +16,9 @@ import torchaudio.transforms as ta_transforms
 from torch import nn
 
 from braindecode.models.base import EEGModuleMixin
-from braindecode.modules import TDSConvEncoder
 
 
-class EMG2QwertyNet(EEGModuleMixin, nn.Module, license="cc-by-nc-sa-4.0"):
+class EMG2QwertyNet(EEGModuleMixin, nn.Module):
     r"""Decoder mapping surface electromyography (sEMG) to keystrokes
     (emg2qwerty) [emg2qwerty2024]_.
 
@@ -91,7 +89,7 @@ class EMG2QwertyNet(EEGModuleMixin, nn.Module, license="cc-by-nc-sa-4.0"):
     .. warning::
         **License: noncommercial use only.** This module is a derivative
         of Meta's reference implementation released under
-        `CC BY-NC-SA 4.0 <https://creativecommons.org/licenses/by-nc-sa/4.0/>`_,
+        `CC BY-NC 4.0 <https://creativecommons.org/licenses/by-nc/4.0/>`_,
         the same license as the upstream repository. Not covered by
         braindecode's BSD-3 license. Must not be used in commercial
         products or services. Pretrained weights from the upstream release
@@ -347,7 +345,7 @@ class EMG2QwertyNet(EEGModuleMixin, nn.Module, license="cc-by-nc-sa-4.0"):
                 activation=activation,
             ),
             nn.Flatten(start_dim=2),
-            TDSConvEncoder(
+            _TDSConvEncoder(
                 num_features=num_features,
                 block_channels=list(block_channels),
                 kernel_width=kernel_width,
@@ -358,7 +356,9 @@ class EMG2QwertyNet(EEGModuleMixin, nn.Module, license="cc-by-nc-sa-4.0"):
 
         self.final_layer = nn.Linear(num_features, self.n_outputs)
 
-        self._n_conv_blocks = len(block_channels)
+        self._n_conv_blocks = sum(
+            isinstance(m, _TDSConv2dBlock) for m in self.model[3].tds_conv_blocks
+        )
 
     def forward(
         self, x: torch.Tensor, return_features: bool = False
@@ -452,11 +452,22 @@ class EMG2QwertyNet(EEGModuleMixin, nn.Module, license="cc-by-nc-sa-4.0"):
         captured init config (``get_config()``) is also kept in sync so
         save/load round-trips rebuild the new head.
         """
-        self._set_n_outputs(n_outputs)
+        if n_outputs <= 0:
+            raise ValueError(f"n_outputs must be positive; got {n_outputs}.")
         old_head = self.final_layer
         self.final_layer = nn.Linear(old_head.in_features, n_outputs).to(
             device=old_head.weight.device, dtype=old_head.weight.dtype
         )
+        self._n_outputs = n_outputs
+        # Keep the init-kwargs snapshot used by ``get_config()`` aligned with
+        # the live head, so ``EMG2QwertyNet.from_config(m.get_config())``
+        # rebuilds the head with the new vocab size.
+        init_kwargs = getattr(self, "_braindecode_init_kwargs", None)
+        if init_kwargs is not None and "n_outputs" in init_kwargs:
+            init_kwargs["n_outputs"] = n_outputs
+        hub_config = getattr(self, "_hub_mixin_config", None)
+        if hub_config is not None and "n_outputs" in hub_config:
+            hub_config["n_outputs"] = n_outputs
 
     def compute_output_lengths(self, input_lengths: torch.Tensor) -> torch.Tensor:
         """Map per-sample input lengths to CTC emission lengths.
@@ -788,3 +799,116 @@ class _MultiBandRotationInvariantMLP(nn.Module):
         for band_idx, mlp in enumerate(self.mlps):
             band_outputs.append(mlp(per_band_inputs[band_idx]))
         return torch.stack(band_outputs, dim=self.stack_dim)
+
+
+class _TDSConv2dBlock(nn.Module):
+    r"""Time-Depth-Separable 2-D conv block ([hannun2019tds]_).
+
+    1×``kernel_width`` conv + activation + residual + LayerNorm. No
+    temporal padding: strips ``kernel_width - 1`` frames per block.
+    """
+
+    def __init__(
+        self,
+        channels: int,
+        width: int,
+        kernel_width: int,
+        activation: type[nn.Module] = nn.ReLU,
+    ) -> None:
+        super().__init__()
+        self.channels = channels
+        self.width = width
+        self.conv2d = nn.Conv2d(
+            in_channels=channels,
+            out_channels=channels,
+            kernel_size=(1, kernel_width),
+        )
+        self.activation = activation()
+        self.layer_norm = nn.LayerNorm(channels * width)
+
+    def forward(self, inputs: torch.Tensor) -> torch.Tensor:
+        n_in_frames, batch_size, n_features = inputs.shape
+        folded = inputs.movedim(0, -1).reshape(
+            batch_size, self.channels, self.width, n_in_frames
+        )
+        conv_out = self.activation(self.conv2d(folded))
+        conv_out = conv_out.reshape(batch_size, n_features, -1).movedim(-1, 0)
+        n_out_frames = conv_out.shape[0]
+        # Output frame i aligns with input frame i+(kernel_width-1), so
+        # the residual is the LAST n_out_frames input frames.
+        residual_added = conv_out + inputs[-n_out_frames:]
+        return self.layer_norm(residual_added)
+
+
+class _TDSFullyConnectedBlock(nn.Module):
+    r"""Two-layer FFN with residual skip, LayerNorm, and optional dropout."""
+
+    def __init__(
+        self,
+        num_features: int,
+        activation: type[nn.Module] = nn.ReLU,
+        drop_prob: float = 0.0,
+    ) -> None:
+        super().__init__()
+        layers: list[nn.Module] = [
+            nn.Linear(num_features, num_features),
+            activation(),
+        ]
+        if drop_prob > 0:
+            layers.append(nn.Dropout(drop_prob))
+        layers.append(nn.Linear(num_features, num_features))
+        if drop_prob > 0:
+            layers.append(nn.Dropout(drop_prob))
+        self.fc_block = nn.Sequential(*layers)
+        self.layer_norm = nn.LayerNorm(num_features)
+
+    def forward(self, inputs: torch.Tensor) -> torch.Tensor:
+        return self.layer_norm(self.fc_block(inputs) + inputs)
+
+
+class _TDSConvEncoder(nn.Module):
+    r"""``_TDSConv2dBlock`` + ``_TDSFullyConnectedBlock`` stack.
+
+    With ``K = len(block_channels)`` blocks and no temporal padding, the
+    encoder strips ``K * (kernel_width - 1)`` frames in total. Each ``ch``
+    must evenly divide ``num_features`` so the conv block can fold features
+    as ``(channels=ch, width=num_features // ch)``.
+    """
+
+    def __init__(
+        self,
+        num_features: int,
+        block_channels: Sequence[int] = (24, 24, 24, 24),
+        kernel_width: int = 32,
+        activation: type[nn.Module] = nn.ReLU,
+        drop_prob: float = 0.0,
+    ) -> None:
+        super().__init__()
+        if not block_channels:
+            raise ValueError("block_channels must be non-empty.")
+        blocks: list[nn.Module] = []
+        for n_block_channels in block_channels:
+            if num_features % n_block_channels != 0:
+                raise ValueError(
+                    f"block_channels entry {n_block_channels} does not evenly "
+                    f"divide num_features={num_features}."
+                )
+            blocks.extend(
+                [
+                    _TDSConv2dBlock(
+                        n_block_channels,
+                        num_features // n_block_channels,
+                        kernel_width,
+                        activation=activation,
+                    ),
+                    _TDSFullyConnectedBlock(
+                        num_features,
+                        activation=activation,
+                        drop_prob=drop_prob,
+                    ),
+                ]
+            )
+        self.tds_conv_blocks = nn.Sequential(*blocks)
+
+    def forward(self, inputs: torch.Tensor) -> torch.Tensor:
+        return self.tds_conv_blocks(inputs)

{braindecode-1.8.0.dev174412347 → braindecode-1.8.0.dev176127673}/braindecode/models/ifnet.py

@@ -1,4 +1,3 @@
-# License: MIT
 """IFNet Neural Network.
 
 Authors: Jiaheng Wang
@@ -29,7 +28,7 @@ from braindecode.modules import (
 )
 
 
-class IFNet(EEGModuleMixin, nn.Module, license="mit"):
+class IFNet(EEGModuleMixin, nn.Module):
     r"""IFNetV2 from Wang J et al (2023) [ifnet]_.
 
     :bdg-success:`Convolution` :bdg-primary:`Filterbank`

{braindecode-1.8.0.dev174412347 → braindecode-1.8.0.dev176127673}/braindecode/models/luna.py

@@ -1,4 +1,3 @@
-# License: Apache-2.0
 """This implementation is adapted from ETH Zurich's BioFoundation repository.
 
 Döner, B., Ingolfsson, T. M., Benini, L., & Li, Y. (2025).
@@ -27,7 +26,7 @@ from braindecode.models.util import extract_channel_locations_from_chs_info
 from braindecode.modules.layers import DropPath
 
 
-class LUNA(EEGModuleMixin, nn.Module, license="apache-2.0"):
+class LUNA(EEGModuleMixin, nn.Module):
     r"""LUNA from Döner et al [LUNA]_.
 
     :bdg-success:`Convolution` :bdg-danger:`Foundation Model` :bdg-dark-line:`Channel`

{braindecode-1.8.0.dev174412347 → braindecode-1.8.0.dev176127673}/braindecode/models/medformer.py

@@ -17,7 +17,7 @@ from braindecode.functional import sinusoidal_positional_encoding
 from braindecode.models.base import EEGModuleMixin
 
 
-class MEDFormer(EEGModuleMixin, nn.Module, license="mit"):
+class MEDFormer(EEGModuleMixin, nn.Module):
     r"""
     Medformer from Wang et al (2024) [Medformer2024]_.
 

{braindecode-1.8.0.dev174412347 → braindecode-1.8.0.dev176127673}/braindecode/models/meta_neuromotor.py

@@ -31,7 +31,7 @@ _LPadType = int | Literal["none", "steady", "full"]
 # ---------------------------------------------------------------------------
 
 
-class MetaNeuromotorHand(EEGModuleMixin, nn.Module, license="cc-by-nc-4.0"):
+class MetaNeuromotorHand(EEGModuleMixin, nn.Module):
     r"""Generic neuromotor interface for handwriting from Meta (2025) [gni2025]_.
 
     :bdg-info:`Attention/Transformer` :bdg-success:`Convolution`

{braindecode-1.8.0.dev174412347 → braindecode-1.8.0.dev176127673}/braindecode/models/mvpformer.py

@@ -26,7 +26,7 @@ from braindecode.models.base import EEGModuleMixin
 from braindecode.modules import PatchTokenizer
 
 
-class MVPFormer(EEGModuleMixin, nn.Module, license="apache-2.0"):
+class MVPFormer(EEGModuleMixin, nn.Module):
     r"""MVPFormer from Carzaniga et al. (2026) [Carzaniga2026]_.
 
     :bdg-danger:`Foundation Model` :bdg-info:`Attention/Transformer`

{braindecode-1.8.0.dev174412347 → braindecode-1.8.0.dev176127673}/braindecode/models/sstdpn.py

@@ -216,7 +216,7 @@ class SSTDPN(EEGModuleMixin, nn.Module):
         temporal dual prototype network for motor imagery
         brain–computer interface. Knowledge-Based Systems,
         315, 113315. GitHub repository.
-        https://github.com/hancan16/SST-DPN
+        https://github.com/hancan16/SST-DPN.
     """
 
     def __init__(

{braindecode-1.8.0.dev174412347 → braindecode-1.8.0.dev176127673}/braindecode/models/summary.csv

@@ -41,9 +41,6 @@ FBCNet,Motor Imagery,Prediction,250,"n_chans, n_outputs, n_times, sfreq",11812,"
 FBMSNet,Motor Imagery,Prediction,250,"n_chans, n_outputs, n_times, sfreq",16231,"FBMSNet(n_chans=22, n_outputs=4, n_times=1000, sfreq=250)","Convolution,FilterBank",EEG
 MetaNeuromotorHand,Handwriting (sEMG),CTC Sequence,2000,"n_chans, n_outputs, n_times, sfreq",1021284,"MetaNeuromotorHand(n_chans=16, n_outputs=100, n_times=32000, sfreq=2000)","Convolution,Attention/Transformer",sEMG
 EMG2QwertyNet,Touch typing (sEMG),CTC Sequence,2000,"n_chans, n_outputs, n_times, sfreq",5293315,"EMG2QwertyNet(n_chans=32, n_outputs=99, n_times=8000, sfreq=2000)","Convolution",sEMG
-VEMG2Pose,Hand pose (sEMG),Sequence Regression,2000,"n_chans, n_outputs, n_times, sfreq",5980328,"VEMG2Pose(n_chans=16, n_outputs=20, n_times=11790, sfreq=2000)","Convolution,Recurrent",sEMG
-NeuroPose,Hand pose (sEMG),Sequence Regression,2000,"n_chans, n_outputs, n_times, sfreq",6354903,"NeuroPose(n_chans=16, n_outputs=20, n_times=10000, sfreq=2000)","Convolution",sEMG
-SensingDynamics,Hand pose (sEMG),Sequence Regression,2000,"n_chans, n_outputs, n_times, sfreq",1738297,"SensingDynamics(n_chans=16, n_outputs=20, n_times=480, sfreq=2000)","Convolution,Channel",sEMG
 FBLightConvNet,Motor Imagery,Prediction,250,"n_chans, n_outputs, n_times, sfreq",6596,"FBLightConvNet(n_chans=22, n_outputs=4, n_times=1000, sfreq=250)","Convolution,FilterBank",EEG
 IFNet,Motor Imagery,Prediction,250,"n_chans, n_outputs, n_times, sfreq",9860,"IFNet(n_chans=22, n_outputs=4, n_times=1000, sfreq=250)","Convolution,FilterBank",EEG
 BrainModule,Speech Decoding,Prediction,250,"n_chans, n_outputs, n_times, sfreq",6186909,"BrainModule(n_chans=64, n_outputs=29, n_times=160, sfreq=1000)","Convolution","EEG, MEG"

{braindecode-1.8.0.dev174412347 → braindecode-1.8.0.dev176127673}/braindecode/models/tcformer.py

@@ -14,7 +14,7 @@ from braindecode.models.base import EEGModuleMixin
 from braindecode.modules import CausalConv1d, Conv1dWithConstraint, DropPath
 
 
-class TCFormer(EEGModuleMixin, nn.Module, license="mit"):
+class TCFormer(EEGModuleMixin, nn.Module):
     r"""TCFormer from Altaheri et al (2025) [tcformer]_.
 
     :bdg-success:`Convolution` :bdg-info:`Attention/Transformer`

{braindecode-1.8.0.dev174412347 → braindecode-1.8.0.dev176127673}/braindecode/models/util.py

@@ -440,39 +440,6 @@ models_mandatory_parameters: list[
             "n_outputs": 99,
         },
     ),
-    (
-        "VEMG2Pose",
-        ["n_chans", "n_outputs", "n_times", "sfreq"],
-        {
-            "n_chans": 16,
-            "n_times": 11790,
-            "sfreq": 2000.0,
-            "input_window_seconds": 5.895,
-            "n_outputs": 20,
-        },
-    ),
-    (
-        "NeuroPose",
-        ["n_chans", "n_outputs", "n_times", "sfreq"],
-        {
-            "n_chans": 16,
-            "n_times": 10000,
-            "sfreq": 2000.0,
-            "input_window_seconds": 5.0,
-            "n_outputs": 20,
-        },
-    ),
-    (
-        "SensingDynamics",
-        ["n_chans", "n_outputs", "n_times", "sfreq"],
-        {
-            "n_chans": 16,
-            "n_times": 480,
-            "sfreq": 2000.0,
-            "input_window_seconds": 480 / 2000.0,
-            "n_outputs": 20,
-        },
-    ),
     ("FBLightConvNet", ["n_chans", "n_outputs", "n_times", "sfreq"], {"sfreq": 200.0}),
     ("IFNet", ["n_chans", "n_outputs", "n_times", "sfreq"], {"sfreq": 200.0}),
     ("PBT", ["n_chans", "n_outputs", "n_times"], None),
@@ -559,10 +526,6 @@ non_classification_models = [
     # Emits a (batch, T_out, vocab) sequence for CTC, not class logits.
     "MetaNeuromotorHand",
     "EMG2QwertyNet",
-    # Dense per-frame pose sequences (batch, T, n_joints), not logits.
-    "VEMG2Pose",
-    "NeuroPose",
-    "SensingDynamics",
     # forward returns (batch, num_latents, n_outputs) dense per-token logits,
     # not class logits.
     "DANCE",

{braindecode-1.8.0.dev174412347 → braindecode-1.8.0.dev176127673}/braindecode/models/zuna.py

@@ -19,7 +19,7 @@ from braindecode.models.util import extract_channel_locations_from_chs_info
 from braindecode.modules import PatchTokenizer
 
 
-class ZUNA(EEGModuleMixin, nn.Module, license="apache-2.0"):
+class ZUNA(EEGModuleMixin, nn.Module):
     r"""ZUNA from Warner et al. (2026) [Warner2026ZUNA]_.
 
     :bdg-danger:`Foundation Model` :bdg-dark-line:`Channel` :bdg-info:`Attention/Transformer`

{braindecode-1.8.0.dev174412347 → braindecode-1.8.0.dev176127673}/braindecode/modules/__init__.py

@@ -1,10 +1,4 @@
-from .activation import (
-    GatedLinearUnit,
-    LogActivation,
-    SafeLog,
-    SmoothMaximumUnit,
-    Square,
-)
+from .activation import GatedLinearUnit, LogActivation, SafeLog, Square
 from .attention import (
     CAT,
     CBAM,
@@ -20,13 +14,7 @@ from .attention import (
     MultiHeadAttention,
     SqueezeAndExcitation,
 )
-from .blocks import (
-    MLP,
-    FeedForwardBlock,
-    InceptionBlock,
-    PatchTokenizer,
-    TDSConvEncoder,
-)
+from .blocks import MLP, FeedForwardBlock, InceptionBlock, PatchTokenizer
 from .convolution import (
     AvgPool2dWithConv,
     CausalConv1d,
@@ -66,7 +54,6 @@ __all__ = [
     "GatedLinearUnit",
     "LogActivation",
     "SafeLog",
-    "SmoothMaximumUnit",
     "Square",
     "CAT",
     "CBAM",
@@ -86,7 +73,6 @@ __all__ = [
     "FeedForwardBlock",
     "InceptionBlock",
     "PatchTokenizer",
-    "TDSConvEncoder",
     "AvgPool2dWithConv",
     "CausalConv1d",
     "CombinedConv",