Braindecode 1.8.0.dev170432067__tar.gz → 1.8.0.dev173646312__tar.gz

braindecode-1.8.0.dev173646312/NOTICE.txt

@@ -0,0 +1,65 @@
+# BRAINDECODE Notice
+
+## Licensed Components
+
+### BSD-3-Clause Licensed Files
+
+All files within the `braindecode/` package are licensed under the BSD-3-Clause
+License, except for those listed in the sections below.
+
+### 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`
+- `braindecode/models/brainmodule.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.
+
+### CC BY-NC-SA 4.0 Licensed Files
+
+The following components are licensed under the Creative Commons
+Attribution-NonCommercial-ShareAlike 4.0 International License:
+
+- `braindecode/models/emg2qwerty.py`
+
+The `emg2qwerty.py` file is a derivative of `facebookresearch/emg2qwerty`,
+released by Meta Platforms, Inc. under CC BY-NC-SA 4.0, and inherits the
+same noncommercial ShareAlike terms.
+
+### MIT Licensed Files
+
+The following components are licensed under the MIT License:
+
+- `braindecode/models/ctnet.py`
+- `braindecode/models/dance.py`
+- `braindecode/models/medformer.py`
+- `braindecode/models/tcformer.py`
+- `braindecode/models/ifnet.py`
+
+### Apache-2.0 Licensed Files
+
+The following components are licensed under the Apache License 2.0:
+
+- `braindecode/models/mvpformer.py`
+- `braindecode/models/zuna.py`
+- `braindecode/models/luna.py`
+
+## 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/)
+- [CC BY-NC-SA 4.0 License](https://creativecommons.org/licenses/by-nc-sa/4.0/)
+- [MIT License](https://opensource.org/licenses/MIT)
+- [Apache-2.0 License](https://www.apache.org/licenses/LICENSE-2.0)
+
+## Note
+
+This list covers files whose own headers declare a license other than
+BSD-3-Clause. Per-file provenance review of the remaining models is
+tracked separately.

{braindecode-1.8.0.dev170432067/braindecode.egg-info → braindecode-1.8.0.dev173646312}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: braindecode
-Version: 1.8.0.dev170432067
+Version: 1.8.0.dev173646312
 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>
@@ -260,7 +260,10 @@ This project is primarily licensed under the BSD-3-Clause License.
 Additional Components
 =====================
 
-Some components within this repository are licensed under the Creative Commons
-Attribution-NonCommercial 4.0 International License.
+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.
 
-Please refer to the ``LICENSE`` and ``NOTICE`` files for more detailed information.
+Please refer to the ``LICENSE`` and ``NOTICE`` files for the per-file list and more
+detailed information.

{braindecode-1.8.0.dev170432067 → braindecode-1.8.0.dev173646312}/README.rst

@@ -170,7 +170,10 @@ This project is primarily licensed under the BSD-3-Clause License.
 Additional Components
 =====================
 
-Some components within this repository are licensed under the Creative Commons
-Attribution-NonCommercial 4.0 International License.
+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.
 
-Please refer to the ``LICENSE`` and ``NOTICE`` files for more detailed information.
+Please refer to the ``LICENSE`` and ``NOTICE`` files for the per-file list and more
+detailed information.

{braindecode-1.8.0.dev170432067 → braindecode-1.8.0.dev173646312}/braindecode/functional/functions.py

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

{braindecode-1.8.0.dev170432067 → braindecode-1.8.0.dev173646312}/braindecode/models/base.py

@@ -519,6 +519,31 @@ 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.dev170432067 → braindecode-1.8.0.dev173646312}/braindecode/models/brainmodule.py

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

{braindecode-1.8.0.dev170432067 → braindecode-1.8.0.dev173646312}/braindecode/models/ctnet.py

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

{braindecode-1.8.0.dev170432067 → braindecode-1.8.0.dev173646312}/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):
+class DANCE(EEGModuleMixin, nn.Module, license="mit"):
     r"""DANCE from Lévy et al (2026) [dance]_.
 
     :bdg-success:`Convolution` :bdg-info:`Attention/Transformer` :bdg-dark-line:`Channel`
@@ -349,8 +349,7 @@ class DANCE(EEGModuleMixin, nn.Module):
 
     def reset_head(self, n_outputs: int) -> None:
         """Replace the dense head and the DETR class head for a new ``n_outputs``."""
-        if n_outputs <= 0:
-            raise ValueError(f"n_outputs must be positive; got {n_outputs}.")
+        self._set_n_outputs(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
@@ -359,10 +358,3 @@ class DANCE(EEGModuleMixin, nn.Module):
         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.dev170432067 → braindecode-1.8.0.dev173646312}/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.dev170432067 → braindecode-1.8.0.dev173646312}/braindecode/models/eegminer.py

@@ -6,6 +6,7 @@
 """
 
 # 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
@@ -48,7 +49,7 @@ class _EEGMinerFeatures(nn.Module):
         return x[:, self.triu_row, self.triu_col, :]
 
 
-class EEGMiner(EEGModuleMixin, nn.Module):
+class EEGMiner(EEGModuleMixin, nn.Module, license="cc-by-nc-4.0"):
     r"""EEGMiner from Ludwig et al (2024) [eegminer]_.
 
     :bdg-success:`Convolution` :bdg-warning:`Interpretability`

{braindecode-1.8.0.dev170432067 → braindecode-1.8.0.dev173646312}/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.dev170432067 → braindecode-1.8.0.dev173646312}/braindecode/models/emg2qwerty.py

@@ -1,9 +1,10 @@
 # Authors: Meta Platforms, Inc. and affiliates (original emg2qwerty)
 #          Bruno Aristimunha <b.aristimunha@gmail.com> (Braindecode adaptation)
 #
-# License: Creative Commons Attribution-NonCommercial 4.0 International (CC BY-NC 4.0)
+# License: Creative Commons Attribution-NonCommercial-ShareAlike 4.0
+#          International (CC BY-NC-SA 4.0)
 # Adapted from https://github.com/facebookresearch/emg2qwerty (NeurIPS 2024).
-# Inherits CC BY-NC 4.0; not covered by braindecode's BSD-3 license.
+# Inherits CC BY-NC-SA 4.0; not covered by braindecode's BSD-3 license.
 """``EMG2QwertyNet``: TDS-Conv-CTC for sEMG-to-keystroke decoding."""
 
 from __future__ import annotations
@@ -16,9 +17,10 @@ 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):
+class EMG2QwertyNet(EEGModuleMixin, nn.Module, license="cc-by-nc-sa-4.0"):
     r"""Decoder mapping surface electromyography (sEMG) to keystrokes
     (emg2qwerty) [emg2qwerty2024]_.
 
@@ -89,7 +91,7 @@ class EMG2QwertyNet(EEGModuleMixin, nn.Module):
     .. warning::
         **License: noncommercial use only.** This module is a derivative
         of Meta's reference implementation released under
-        `CC BY-NC 4.0 <https://creativecommons.org/licenses/by-nc/4.0/>`_,
+        `CC BY-NC-SA 4.0 <https://creativecommons.org/licenses/by-nc-sa/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
@@ -345,7 +347,7 @@ class EMG2QwertyNet(EEGModuleMixin, nn.Module):
                 activation=activation,
             ),
             nn.Flatten(start_dim=2),
-            _TDSConvEncoder(
+            TDSConvEncoder(
                 num_features=num_features,
                 block_channels=list(block_channels),
                 kernel_width=kernel_width,
@@ -356,9 +358,7 @@ class EMG2QwertyNet(EEGModuleMixin, nn.Module):
 
         self.final_layer = nn.Linear(num_features, self.n_outputs)
 
-        self._n_conv_blocks = sum(
-            isinstance(m, _TDSConv2dBlock) for m in self.model[3].tds_conv_blocks
-        )
+        self._n_conv_blocks = len(block_channels)
 
     def forward(
         self, x: torch.Tensor, return_features: bool = False
@@ -452,22 +452,11 @@ class EMG2QwertyNet(EEGModuleMixin, nn.Module):
         captured init config (``get_config()``) is also kept in sync so
         save/load round-trips rebuild the new head.
         """
-        if n_outputs <= 0:
-            raise ValueError(f"n_outputs must be positive; got {n_outputs}.")
+        self._set_n_outputs(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.
@@ -799,116 +788,3 @@ 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.dev170432067 → braindecode-1.8.0.dev173646312}/braindecode/models/ifnet.py

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

{braindecode-1.8.0.dev170432067 → braindecode-1.8.0.dev173646312}/braindecode/models/luna.py

@@ -1,3 +1,4 @@
+# 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).
@@ -26,7 +27,7 @@ from braindecode.models.util import extract_channel_locations_from_chs_info
 from braindecode.modules.layers import DropPath
 
 
-class LUNA(EEGModuleMixin, nn.Module):
+class LUNA(EEGModuleMixin, nn.Module, license="apache-2.0"):
     r"""LUNA from Döner et al [LUNA]_.
 
     :bdg-success:`Convolution` :bdg-danger:`Foundation Model` :bdg-dark-line:`Channel`

{braindecode-1.8.0.dev170432067 → braindecode-1.8.0.dev173646312}/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):
+class MEDFormer(EEGModuleMixin, nn.Module, license="mit"):
     r"""
     Medformer from Wang et al (2024) [Medformer2024]_.
 

{braindecode-1.8.0.dev170432067 → braindecode-1.8.0.dev173646312}/braindecode/models/meta_neuromotor.py

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

{braindecode-1.8.0.dev170432067 → braindecode-1.8.0.dev173646312}/braindecode/models/mvpformer.py

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

{braindecode-1.8.0.dev170432067 → braindecode-1.8.0.dev173646312}/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.dev170432067 → braindecode-1.8.0.dev173646312}/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):
+class TCFormer(EEGModuleMixin, nn.Module, license="mit"):
     r"""TCFormer from Altaheri et al (2025) [tcformer]_.
 
     :bdg-success:`Convolution` :bdg-info:`Attention/Transformer`

{braindecode-1.8.0.dev170432067 → braindecode-1.8.0.dev173646312}/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):
+class ZUNA(EEGModuleMixin, nn.Module, license="apache-2.0"):
     r"""ZUNA from Warner et al. (2026) [Warner2026ZUNA]_.
 
     :bdg-danger:`Foundation Model` :bdg-dark-line:`Channel` :bdg-info:`Attention/Transformer`

{braindecode-1.8.0.dev170432067 → braindecode-1.8.0.dev173646312}/braindecode/modules/__init__.py

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