Braindecode 1.8.0__tar.gz → 1.8.0.dev1058__tar.gz

braindecode-1.8.0.dev1058/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/braindecode.egg-info → braindecode-1.8.0.dev1058}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: braindecode
-Version: 1.8.0
+Version: 1.8.0.dev1058
 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>
@@ -34,6 +34,7 @@ Requires-Dist: pandas
 Requires-Dist: wfdb>=4.3.1
 Requires-Dist: linear_attention_transformer
 Requires-Dist: docstring_inheritance
+Requires-Dist: rotary_embedding_torch
 Requires-Dist: pydantic>=2.0
 Provides-Extra: moabb
 Requires-Dist: moabb>=1.4.3; extra == "moabb"
@@ -50,7 +51,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 +260,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 → braindecode-1.8.0.dev1058}/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 → braindecode-1.8.0.dev1058}/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
@@ -60,7 +59,6 @@ def _html_row(label, value):
 
 _METADATA_INTERNAL_COLS = {
     "i_window_in_trial",
-    "i_trial_in_dataset",
     "i_start_in_trial",
     "i_stop_in_trial",
     "target",
@@ -1175,8 +1173,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__(
@@ -1378,20 +1374,6 @@ class BaseConcatDataset(ConcatDataset, HubDatasetMixin, Generic[T]):
                 "datasets are WindowsDataset."
             )
 
-        for ds in self.datasets:
-            if hasattr(ds, "_windows") and ds._windows is not None:
-                df = ds._windows.metadata
-            else:
-                df = ds.metadata
-            if (
-                "i_trial_in_dataset" in df.columns
-                and "i_trial_in_dataset" in ds.description
-            ):
-                raise ValueError(
-                    "Dataset descriptions cannot contain the reserved window "
-                    "metadata key 'i_trial_in_dataset'."
-                )
-
         all_dfs = list()
         for ds in self.datasets:
             if hasattr(ds, "_windows") and ds._windows is not None:

{braindecode-1.8.0 → braindecode-1.8.0.dev1058}/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 → braindecode-1.8.0.dev1058}/braindecode/models/atcnet.py

@@ -196,7 +196,7 @@ class ATCNet(EEGModuleMixin, nn.Module):
     num_heads : int
         Number of attention heads, denoted H in table 1 of the paper [1]_.
         Defaults to 2 as in [1]_.
-    att_drop_prob : float
+    att_dropout : float
         Dropout probability used in the attention block, denoted pa in table 1
         of the paper [1]_. Defaults to 0.5 as in [1]_.
     tcn_depth : int
@@ -206,7 +206,7 @@ class ATCNet(EEGModuleMixin, nn.Module):
     tcn_kernel_size : int
         Temporal kernel size used in TCN block, denoted Kt in table 1 of the
         paper [1]_. Defaults to 4 as in [1]_.
-    tcn_drop_prob : float
+    tcn_dropout : float
         Dropout probability used in the TCN block, denoted pt in table 1
         of the paper [1]_. Defaults to 0.3 as in [1]_.
     tcn_activation : torch.nn.Module

{braindecode-1.8.0 → braindecode-1.8.0.dev1058}/braindecode/models/attn_sleep.py

@@ -1,36 +1,10 @@
 # Authors: Divyesh Narayanan <divyesh.narayanan@gmail.com>
-#          Sarthak Tayal <sarthaktayal2@gmail.com>
 #
 # License: BSD (3-clause)
-#
-# This implementation derives from https://github.com/emadeldeen24/AttnSleep:
-#
-# MIT License
-#
-# Copyright (c) 2020 Emadeldeen Eldele
-#
-# Permission is hereby granted, free of charge, to any person obtaining a copy
-# of this software and associated documentation files (the "Software"), to deal
-# in the Software without restriction, including without limitation the rights
-# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-# copies of the Software, and to permit persons to whom the Software is
-# furnished to do so, subject to the following conditions:
-#
-# The above copyright notice and this permission notice shall be included in all
-# copies or substantial portions of the Software.
-#
-# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-# SOFTWARE.
 
 import math
 import warnings
 from copy import deepcopy
-from numbers import Integral
 
 import torch
 import torch.nn.functional as F
@@ -59,8 +33,7 @@ class AttnSleep(EEGModuleMixin, nn.Module):
 
     Warning - This model was designed for signals of 30 seconds at 100Hz or 125Hz (in which case
     the reference architecture from [1]_ which was validated on SHHS dataset [2]_ will be used)
-    to use any other input is likely to make the model perform in unintended ways. Any other
-    window length also needs a ``d_model`` of its own, see the parameter below.
+    to use any other input is likely to make the model perform in unintended ways.
 
     Parameters
     ----------
@@ -71,10 +44,7 @@ class AttnSleep(EEGModuleMixin, nn.Module):
         Also the input dimension of the first FC layer in the feed forward
         and the output of the second FC layer in the same.
         Increase for higher sampling rate/signal length.
-        It should be divisible by n_attn_heads. It must also equal the number
-        of time steps returned by the feature extractor: 80 for 30 seconds at
-        100 Hz and 100 for 30 seconds at 125 Hz. A construction error reports
-        the value needed for other window lengths.
+        It should be divisible by n_attn_heads
     d_ff : int
         Output dimension of the first FC layer in the feed forward and the
         input dimension of the second FC layer in the same.
@@ -88,12 +58,15 @@ class AttnSleep(EEGModuleMixin, nn.Module):
         If True, return the features, i.e. the output of the feature extractor
         (before the final linear layer). If False, pass the features through
         the final linear layer.
+    n_classes : int
+        Alias for `n_outputs`.
+    input_size_s : float
+        Alias for `input_window_seconds`.
     activation : nn.Module, default=nn.ReLU
-        Activation function class to apply in the AFR block and the TCE
-        feed-forward block. Should be a PyTorch activation
+        Activation function class to apply. Should be a PyTorch activation
         module class like ``nn.ReLU`` or ``nn.ELU``. Default is ``nn.ReLU``.
-    activation_mrcnn : nn.Module, default=nn.GELU
-        Activation function class to apply in the multi-resolution CNN layer.
+    activation_mrcnn : nn.Module, default=nn.ReLU
+        Activation function class to apply in the Mask R-CNN layer.
         Should be a PyTorch activation module class like ``nn.ReLU`` or
         ``nn.GELU``. Default is ``nn.GELU``.
 
@@ -126,14 +99,6 @@ class AttnSleep(EEGModuleMixin, nn.Module):
         n_chans=None,
         n_times=None,
     ):
-        if (
-            sum(value is not None for value in (n_times, sfreq, input_window_seconds))
-            < 2
-        ):
-            raise ValueError(
-                "AttnSleep requires at least two of n_times, sfreq, and "
-                "input_window_seconds."
-            )
         super().__init__(
             n_outputs=n_outputs,
             n_chans=n_chans,
@@ -175,14 +140,6 @@ class AttnSleep(EEGModuleMixin, nn.Module):
             activation=activation_mrcnn,
             activation_se=activation,
         )
-        feature_length = self._feature_length(mrcnn, self.n_times)
-        if feature_length != d_model:
-            raise ValueError(
-                f"d_model is {d_model} but the feature extractor returns "
-                f"{feature_length} time steps for an input of {self.n_times} "
-                f"samples at {self.sfreq} Hz. Set d_model={feature_length}, with "
-                "an n_attn_heads that divides it."
-            )
         attn = _MultiHeadedAttention(n_attn_heads, d_model, after_reduced_cnn_size)
         ff = _PositionwiseFeedForward(d_model, d_ff, drop_prob, activation=activation)
         tce = _TCE(
@@ -193,26 +150,23 @@ class AttnSleep(EEGModuleMixin, nn.Module):
         )
 
         self.feature_extractor = nn.Sequential(mrcnn, tce)
-        self.len_last_layer = feature_length * after_reduced_cnn_size
+        self.len_last_layer = self._len_last_layer(self.n_times)
         self.return_feats = return_feats
 
         # TODO: Add new way to handle return features
         """if return_feats:
             raise ValueError("return_feat == True is not accepted anymore")"""
         if not return_feats:
-            self.final_layer = nn.Linear(self.len_last_layer, self.n_outputs)
-
-    @staticmethod
-    def _feature_length(mrcnn, n_times):
-        training_states = [(module, module.training) for module in mrcnn.modules()]
-        mrcnn.eval()
-        try:
-            with torch.no_grad():
-                out = mrcnn(torch.zeros(1, 1, n_times))
-        finally:
-            for module, was_training in training_states:
-                module.training = was_training
-        return out.shape[-1]
+            self.final_layer = nn.Linear(
+                d_model * after_reduced_cnn_size, self.n_outputs
+            )
+
+    def _len_last_layer(self, input_size):
+        self.feature_extractor.eval()
+        with torch.no_grad():
+            out = self.feature_extractor(torch.Tensor(1, 1, input_size))
+        self.feature_extractor.train()
+        return len(out.flatten())
 
     def forward(self, x: torch.Tensor) -> torch.Tensor:
         """
@@ -241,7 +195,7 @@ class _SELayer(nn.Module):
         self.avg_pool = nn.AdaptiveAvgPool1d(1)
         self.fc = nn.Sequential(
             nn.Linear(channel, channel // reduction, bias=False),
-            activation(),
+            activation(inplace=True),
             nn.Linear(channel // reduction, channel, bias=False),
             nn.Sigmoid(),
         )
@@ -282,10 +236,10 @@ class _SEBasicBlock(nn.Module):
         super(_SEBasicBlock, self).__init__()
         self.conv1 = nn.Conv1d(inplanes, planes, stride)
         self.bn1 = nn.BatchNorm1d(planes)
-        self.relu = activation()
+        self.relu = activation(inplace=True)
         self.conv2 = nn.Conv1d(planes, planes, 1)
         self.bn2 = nn.BatchNorm1d(planes)
-        self.se = _SELayer(planes, reduction, activation=activation)
+        self.se = _SELayer(planes, reduction)
         self.downsample = downsample
         self.stride = stride
         self.features = nn.Sequential(
@@ -366,11 +320,11 @@ class _MRCNN(nn.Module):
         self.dropout = nn.Dropout(drate)
         self.inplanes = 128
         self.AFR = self._make_layer(
-            _SEBasicBlock, after_reduced_cnn_size, 1, activation=activation_se
+            _SEBasicBlock, after_reduced_cnn_size, 1, activate=activation_se
         )
 
     def _make_layer(
-        self, block, planes, blocks, stride=1, activation: type[nn.Module] = nn.ReLU
+        self, block, planes, blocks, stride=1, activate: type[nn.Module] = nn.ReLU
     ):  # makes residual SE block
         downsample = None
         if stride != 1 or self.inplanes != planes * block.expansion:
@@ -386,12 +340,10 @@ class _MRCNN(nn.Module):
             )
 
         layers = []
-        layers.append(
-            block(self.inplanes, planes, stride, downsample, activation=activation)
-        )
+        layers.append(block(self.inplanes, planes, stride, downsample))
         self.inplanes = planes * block.expansion
         for i in range(1, blocks):
-            layers.append(block(self.inplanes, planes, activation=activation))
+            layers.append(block(self.inplanes, planes, activate=activate))
 
         return nn.Sequential(*layers)
 
@@ -423,20 +375,9 @@ class _MultiHeadedAttention(nn.Module):
     def __init__(self, h, d_model, after_reduced_cnn_size, dropout=0.1):
         """Take in model size and number of heads."""
         super().__init__()
-        if (
-            isinstance(h, bool)
-            or not isinstance(h, Integral)
-            or h <= 0
-            or d_model % h != 0
-        ):
-            raise ValueError(
-                "n_attn_heads must be a positive integer that divides d_model, "
-                f"got n_attn_heads={h!r} and d_model={d_model}."
-            )
-        h = int(h)
+        assert d_model % h == 0
         self.d_per_head = d_model // h
         self.h = h
-        self.attn = torch.empty(0)
 
         base_conv = CausalConv1d(
             in_channels=after_reduced_cnn_size,

{braindecode-1.8.0 → braindecode-1.8.0.dev1058}/braindecode/models/base.py

@@ -80,7 +80,7 @@ class _BraindecodeDocstringMeta(NumpyDocstringInheritanceInitMeta):
     unwrapped function and correctly inherits ``cls.__doc__``.
     """
 
-    def __init__(cls, class_name, class_bases, class_dict, **kwargs):
+    def __init__(cls, class_name, class_bases, class_dict):
         super().__init__(class_name, class_bases, class_dict)
         # Only wrap subclass __init__s, not EEGModuleMixin itself.
         # Wrapping the mixin would cause super().__init__() calls to
@@ -214,8 +214,6 @@ class EEGModuleMixin(_BaseHubMixin, metaclass=_BraindecodeDocstringMeta):
     __jit_unused_properties__ = ["chs_info"]
 
     def __init_subclass__(cls, **kwargs):
-        license = kwargs.pop("license", "bsd-3-clause")
-
         # TorchScript only honours ``__jit_ignored_attributes__`` for
         # properties defined directly on the concrete class: it collects them
         # with ``vars(type(module))``, which skips inherited ones. Rebinding
@@ -255,6 +253,8 @@ class EEGModuleMixin(_BaseHubMixin, metaclass=_BraindecodeDocstringMeta):
         )
         repo_url = kwargs.pop("repo_url", "https://braindecode.org")
         library_name = kwargs.pop("library_name", "braindecode")
+        license = kwargs.pop("license", "bsd-3-clause")
+
         # Register a coder so that type[nn.Module] parameters
         # (e.g. activation=nn.ELU) are serialized as importable
         # strings in config.json and decoded back on load.
@@ -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 → braindecode-1.8.0.dev1058}/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 → braindecode-1.8.0.dev1058}/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`
@@ -59,19 +59,16 @@ class CTNet(EEGModuleMixin, nn.Module, license="mit"):
 
     Parameters
     ----------
-    activation_patch : nn.Module, default=nn.ELU
-        Activation function to use in the convolutional patch embedding.
-    activation_transformer : nn.Module, default=nn.GELU
-        Activation function to use in the Transformer encoder.
+    activation : nn.Module, default=nn.GELU
+        Activation function to use in the network.
     num_heads : int, default=4
         Number of attention heads in the Transformer encoder.
-    embed_dim : int or None, default=40
+    embed_dim : int or None, default=None
         Embedding size (dimensionality) for the Transformer encoder.
     num_layers : int, default=6
         Number of encoder layers in the Transformer.
-    n_filters_time : int or None, default=None
-        Number of temporal filters in the first convolutional layer. Inferred
-        from ``embed_dim`` and ``depth_multiplier`` when left at ``None``.
+    n_filters_time : int, default=20
+        Number of temporal filters in the first convolutional layer.
     kernel_size : int, default=64
         Kernel size for the temporal convolutional layer.
     depth_multiplier : int, default=2
@@ -80,7 +77,7 @@ class CTNet(EEGModuleMixin, nn.Module, license="mit"):
         Pooling size for the first average pooling layer.
     pool_size_2 : int, default=8
         Pooling size for the second average pooling layer.
-    cnn_drop_prob : float, default=0.3
+        cnn_drop_prob: float, default=0.3
         Dropout probability after convolutional layers.
     att_positional_drop_prob : float, default=0.1
         Dropout probability for the positional encoding in the Transformer.

{braindecode-1.8.0 → braindecode-1.8.0.dev1058}/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 → braindecode-1.8.0.dev1058}/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 → braindecode-1.8.0.dev1058}/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 → braindecode-1.8.0.dev1058}/braindecode/models/eegsimpleconv.py

@@ -71,19 +71,17 @@ class EEGSimpleConv(EEGModuleMixin, torch.nn.Module):
 
     Parameters
     ----------
-    feature_maps : int
+    feature_maps: int
         Number of Feature Maps at the first Convolution, width of the model.
-    n_convs : int
+    n_convs: int
         Number of blocks of convolutions (2 convolutions per block), depth of the model.
-    resampling_freq : int
+    resampling: int
         Resampling Frequency.
-    kernel_size : int
+    kernel_size: int
         Size of the convolutions kernels.
-    return_feature : bool, default=False
-        If True, return the features before the final linear layer.
-    activation : nn.Module, default=nn.ReLU
+    activation: nn.Module, default=nn.ELU
         Activation function class to apply. Should be a PyTorch activation
-        module class like ``nn.ReLU`` or ``nn.ELU``. Default is ``nn.ReLU``.
+        module class like ``nn.ReLU`` or ``nn.ELU``. Default is ``nn.ELU``.
 
     References
     ----------
@@ -93,7 +91,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.
 
     """