braindecode 1.5.1__tar.gz → 1.5.2__tar.gz

{braindecode-1.5.1/braindecode.egg-info → braindecode-1.5.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: braindecode
-Version: 1.5.1
+Version: 1.5.2
 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.5.1 → braindecode-1.5.2}/braindecode/datasets/bids/hub.py

@@ -466,6 +466,7 @@ class HubDatasetMixin:
         token: Optional[str] = None,
         cache_dir: Optional[Union[str, Path]] = None,
         force_download: bool = False,
+        revision: str | None = None,
         **kwargs,
     ):
         """
@@ -485,6 +486,9 @@ class HubDatasetMixin:
             cache directory (~/.cache/huggingface/datasets).
         force_download : bool, default=False
             Whether to force re-download even if cached.
+        revision : str | None, default=None
+            Specific branch, tag, or commit to download. If None, uses the
+            repository's default revision.
         **kwargs
             Additional arguments (currently unused).
 
@@ -526,6 +530,7 @@ class HubDatasetMixin:
                 token=token,
                 cache_dir=cache_dir,
                 force_download=force_download,
+                revision=revision,
             )
 
             # Load format info

{braindecode-1.5.1 → braindecode-1.5.2}/braindecode/models/contrawr.py

@@ -5,6 +5,7 @@ import torch.nn as nn
 from mne.utils import warn
 
 from braindecode.models.base import EEGModuleMixin
+from braindecode.models.util import _disable_batch_norm_training_if_batch_size_one
 
 
 class ContraWR(EEGModuleMixin, nn.Module):
@@ -122,6 +123,7 @@ class ContraWR(EEGModuleMixin, nn.Module):
         self.activation_layer = activation()
         self.final_layer = nn.Linear(emb_size, self.n_outputs)
 
+    @_disable_batch_norm_training_if_batch_size_one
     def forward(self, X: torch.Tensor) -> torch.Tensor:
         """
         Forward pass.

{braindecode-1.5.1 → braindecode-1.5.2}/braindecode/models/deepsleepnet.py

@@ -7,6 +7,7 @@ import torch.nn as nn
 from einops.layers.torch import Rearrange
 
 from braindecode.models.base import EEGModuleMixin
+from braindecode.models.util import _disable_batch_norm_training_if_batch_size_one
 
 
 class DeepSleepNet(EEGModuleMixin, nn.Module):
@@ -289,6 +290,7 @@ class DeepSleepNet(EEGModuleMixin, nn.Module):
             else nn.Linear(fc_out_features, self.n_outputs)
         )
 
+    @_disable_batch_norm_training_if_batch_size_one
     def forward(self, x):
         if x.ndim == 3:
             x = x.unsqueeze(1)

{braindecode-1.5.1 → braindecode-1.5.2}/braindecode/models/eegminer.py

@@ -13,6 +13,7 @@ from torch import nn
 
 import braindecode.functional as F
 from braindecode.models.base import EEGModuleMixin
+from braindecode.models.util import _disable_batch_norm_training_if_batch_size_one
 from braindecode.modules import GeneralizedGaussianFilter
 
 _eeg_miner_methods = ["mag", "corr", "plv"]
@@ -191,6 +192,7 @@ class EEGMiner(EEGModuleMixin, nn.Module):
         self.final_layer = nn.Linear(self.n_features, self.n_outputs)
         nn.init.zeros_(self.final_layer.bias)
 
+    @_disable_batch_norm_training_if_batch_size_one
     def forward(self, x):
         """x: (batch, electrodes, time)"""
         batch = x.shape[0]

{braindecode-1.5.1 → braindecode-1.5.2}/braindecode/models/eegsym.py

@@ -11,6 +11,7 @@ from braindecode.datautil.channel_utils import (
     match_hemisphere_chans,
 )
 from braindecode.models.base import EEGModuleMixin
+from braindecode.models.util import _disable_batch_norm_training_if_batch_size_one
 
 
 class EEGSym(EEGModuleMixin, nn.Module):
@@ -363,6 +364,7 @@ class EEGSym(EEGModuleMixin, nn.Module):
             out_features=self.n_outputs,
         )
 
+    @_disable_batch_norm_training_if_batch_size_one
     def forward(self, x):
         """Forward pass.
 

{braindecode-1.5.1 → braindecode-1.5.2}/braindecode/models/usleep.py

@@ -9,6 +9,7 @@ import torch
 from torch import nn
 
 from braindecode.models.base import EEGModuleMixin
+from braindecode.models.util import _disable_batch_norm_training_if_batch_size_one
 
 
 class USleep(EEGModuleMixin, nn.Module):
@@ -295,6 +296,7 @@ class USleep(EEGModuleMixin, nn.Module):
             # output is (B, n_classes, S)
         )
 
+    @_disable_batch_norm_training_if_batch_size_one
     def forward(self, x: torch.Tensor) -> torch.Tensor:
         """If input x has shape (B, S, C, T), return y_pred of shape (B, n_classes, S).
         If input x has shape (B, C, T), return y_pred of shape (B, n_classes).

{braindecode-1.5.1 → braindecode-1.5.2}/braindecode/models/util.py

@@ -12,6 +12,8 @@ from typing import Any, Dict, Literal, Optional, Sequence
 import numpy as np
 import pandas as pd
 import pydantic
+import torch  # noqa: F401  # exposed so TorchScript can resolve ``torch`` through the BatchNorm-guard wrapper's __globals__
+from torch import nn
 
 models_dict = {}
 
@@ -23,6 +25,38 @@ _EEG_PARAMS = frozenset(
 )
 
 _JSON_SAFE = (int, float, str, bool, type(None))
+_BATCH_NORM_MODULES = (
+    nn.BatchNorm1d,
+    nn.BatchNorm2d,
+    nn.BatchNorm3d,
+    nn.SyncBatchNorm,
+)
+
+
+def _disable_batch_norm_training_if_batch_size_one(forward):
+    """Temporarily set BatchNorm layers to train(False) for batch size one."""
+    forward_signature = inspect.signature(forward)
+
+    @wraps(forward)
+    def wrapped(self, *args, **kwargs):
+        bound = forward_signature.bind(self, *args, **kwargs)
+        x = next(value for name, value in bound.arguments.items() if name != "self")
+        batch_norms = []
+        if self.training and x.shape[0] == 1:
+            batch_norms = [
+                layer
+                for layer in self.modules()
+                if isinstance(layer, _BATCH_NORM_MODULES) and layer.training
+            ]
+            for batch_norm in batch_norms:
+                batch_norm.train(False)
+        try:
+            return forward(self, *args, **kwargs)
+        finally:
+            for batch_norm in batch_norms:
+                batch_norm.train(True)
+
+    return wrapped
 
 
 def _is_jsonable(val):

braindecode-1.5.2/braindecode/version.py

@@ -0,0 +1 @@
+__version__ = "1.5.2"

{braindecode-1.5.1 → braindecode-1.5.2/braindecode.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: braindecode
-Version: 1.5.1
+Version: 1.5.2
 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.5.2/braindecode.egg-info/SOURCES.txt

@@ -0,0 +1,169 @@
+LICENSE.txt
+MANIFEST.in
+NOTICE.txt
+README.rst
+pyproject.toml
+setup.cfg
+braindecode/__init__.py
+braindecode/classifier.py
+braindecode/eegneuralnet.py
+braindecode/regressor.py
+braindecode/util.py
+braindecode/version.py
+braindecode.egg-info/PKG-INFO
+braindecode.egg-info/SOURCES.txt
+braindecode.egg-info/dependency_links.txt
+braindecode.egg-info/requires.txt
+braindecode.egg-info/top_level.txt
+braindecode/augmentation/__init__.py
+braindecode/augmentation/base.py
+braindecode/augmentation/functional.py
+braindecode/augmentation/transforms.py
+braindecode/datasets/__init__.py
+braindecode/datasets/base.py
+braindecode/datasets/bbci.py
+braindecode/datasets/bcicomp.py
+braindecode/datasets/chb_mit.py
+braindecode/datasets/mne.py
+braindecode/datasets/moabb.py
+braindecode/datasets/nmt.py
+braindecode/datasets/registry.py
+braindecode/datasets/siena.py
+braindecode/datasets/sleep_physio_challe_18.py
+braindecode/datasets/sleep_physionet.py
+braindecode/datasets/tuh.py
+braindecode/datasets/utils.py
+braindecode/datasets/xy.py
+braindecode/datasets/bids/__init__.py
+braindecode/datasets/bids/datasets.py
+braindecode/datasets/bids/format.py
+braindecode/datasets/bids/hub.py
+braindecode/datasets/bids/hub_format.py
+braindecode/datasets/bids/hub_io.py
+braindecode/datasets/bids/hub_validation.py
+braindecode/datasets/bids/iterable.py
+braindecode/datautil/__init__.py
+braindecode/datautil/channel_utils.py
+braindecode/datautil/hub_formats.py
+braindecode/datautil/serialization.py
+braindecode/datautil/util.py
+braindecode/functional/__init__.py
+braindecode/functional/functions.py
+braindecode/functional/initialization.py
+braindecode/models/__init__.py
+braindecode/models/atcnet.py
+braindecode/models/attentionbasenet.py
+braindecode/models/attn_sleep.py
+braindecode/models/base.py
+braindecode/models/bendr.py
+braindecode/models/biot.py
+braindecode/models/brainmodule.py
+braindecode/models/cbramod.py
+braindecode/models/codebrain.py
+braindecode/models/config.py
+braindecode/models/contrawr.py
+braindecode/models/ctnet.py
+braindecode/models/deep4.py
+braindecode/models/deepsleepnet.py
+braindecode/models/dgcnn.py
+braindecode/models/eegconformer.py
+braindecode/models/eeginception_erp.py
+braindecode/models/eeginception_mi.py
+braindecode/models/eegitnet.py
+braindecode/models/eegminer.py
+braindecode/models/eegnet.py
+braindecode/models/eegnex.py
+braindecode/models/eegpt.py
+braindecode/models/eegsimpleconv.py
+braindecode/models/eegsym.py
+braindecode/models/eegtcnet.py
+braindecode/models/emg2qwerty.py
+braindecode/models/fbcnet.py
+braindecode/models/fblightconvnet.py
+braindecode/models/fbmsnet.py
+braindecode/models/hybrid.py
+braindecode/models/ifnet.py
+braindecode/models/interpolated.py
+braindecode/models/labram.py
+braindecode/models/luna.py
+braindecode/models/medformer.py
+braindecode/models/meta_neuromotor.py
+braindecode/models/msvtnet.py
+braindecode/models/patchedtransformer.py
+braindecode/models/reve.py
+braindecode/models/sccnet.py
+braindecode/models/shallow_fbcsp.py
+braindecode/models/signal_jepa.py
+braindecode/models/sinc_shallow.py
+braindecode/models/sleep_stager_blanco_2020.py
+braindecode/models/sleep_stager_chambon_2018.py
+braindecode/models/sparcnet.py
+braindecode/models/sstdpn.py
+braindecode/models/summary.csv
+braindecode/models/syncnet.py
+braindecode/models/tcn.py
+braindecode/models/tidnet.py
+braindecode/models/tsinception.py
+braindecode/models/usleep.py
+braindecode/models/util.py
+braindecode/modules/__init__.py
+braindecode/modules/activation.py
+braindecode/modules/attention.py
+braindecode/modules/blocks.py
+braindecode/modules/convolution.py
+braindecode/modules/filter.py
+braindecode/modules/interpolation.py
+braindecode/modules/layers.py
+braindecode/modules/linear.py
+braindecode/modules/parametrization.py
+braindecode/modules/stats.py
+braindecode/modules/util.py
+braindecode/modules/wrapper.py
+braindecode/preprocessing/__init__.py
+braindecode/preprocessing/eegprep_preprocess.py
+braindecode/preprocessing/mne_preprocess.py
+braindecode/preprocessing/preprocess.py
+braindecode/preprocessing/util.py
+braindecode/preprocessing/windowers.py
+braindecode/samplers/__init__.py
+braindecode/samplers/base.py
+braindecode/samplers/ssl.py
+braindecode/training/__init__.py
+braindecode/training/callbacks.py
+braindecode/training/losses.py
+braindecode/training/scoring.py
+braindecode/visualization/__init__.py
+braindecode/visualization/attribution.py
+braindecode/visualization/confusion_matrices.py
+braindecode/visualization/frequency.py
+braindecode/visualization/metrics.py
+braindecode/visualization/sanity.py
+braindecode/visualization/topology.py
+docs/Makefile
+docs/api.rst
+docs/cite.rst
+docs/conf.py
+docs/help.rst
+docs/index.rst
+docs/sg_execution_times.rst
+docs/whats_new.rst
+docs/_templates/autosummary/class.rst
+docs/_templates/autosummary/class_in_subdir.rst
+docs/_templates/autosummary/function.rst
+docs/_templates/autosummary/function_in_subdir.rst
+docs/install/install.rst
+docs/install/install_pip.rst
+docs/install/install_source.rst
+docs/models/models.rst
+docs/models/models_categorization.rst
+docs/models/models_table.rst
+docs/models/models_visualization.rst
+docs/models/categorization/attention.rst
+docs/models/categorization/channel.rst
+docs/models/categorization/convolution.rst
+docs/models/categorization/filterbank.rst
+docs/models/categorization/gnn.rst
+docs/models/categorization/interpretable.rst
+docs/models/categorization/lbm.rst
+docs/models/categorization/recurrent.rst
+docs/models/categorization/spd.rst

{braindecode-1.5.1 → braindecode-1.5.2}/docs/whats_new.rst

@@ -22,6 +22,51 @@
 .. _current:
 
 
+Current 1.5.2 (2026-05-28)
+===============================
+
+Enhancements
+============
+
+- Add a ``revision`` keyword argument to
+  :meth:`braindecode.datasets.BaseConcatDataset.pull_from_hub` so callers can
+  pin dataset downloads to a specific branch, tag, or commit on the Hugging
+  Face Hub.
+
+API and behavior changes
+========================
+
+- None yet
+
+Requirements
+============
+
+- None yet
+
+Bug fixes
+==========
+
+- Fix :class:`~braindecode.models.ContraWR`,
+  :class:`~braindecode.models.DeepSleepNet`, and
+  :class:`~braindecode.models.EEGMiner` raising a confusing
+  :class:`ValueError` on ``batch_size=1`` forward passes in train mode due to
+  :class:`~torch.nn.BatchNorm1d` (and its 2D/3D variants) requiring more than
+  one sample per channel during training.  The affected BatchNorm layers now
+  use running statistics when ``batch_size == 1`` in train mode.
+  By `Bruno Aristimunha`_.
+
+Code health
+============
+
+- Add a monthly scheduled workflow that cuts a stable PyPI release on
+  the 1st of every month, complementing the existing per-push ``.devN``
+  pipeline. (:gh:`1030` by `Bruno Aristimunha`_)
+
+- Make the monthly release workflow push only the release tag and open a
+  pull request for the version bump, so it never needs to push to the
+  protected ``master`` branch. (:gh:`1031` by `Bruno Aristimunha`_)
+
+
 Current 1.5.1 (stable)
 ===============================
 

braindecode-1.5.1/braindecode/version.py

@@ -1 +0,0 @@
-__version__ = "1.5.1"