braindecode 1.6.0.dev1028__tar.gz → 1.6.1.dev1034__tar.gz

{braindecode-1.6.0.dev1028/braindecode.egg-info → braindecode-1.6.1.dev1034}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: braindecode
-Version: 1.6.0.dev1028
+Version: 1.6.1.dev1034
 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.6.0.dev1028 → braindecode-1.6.1.dev1034}/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.6.0.dev1028 → braindecode-1.6.1.dev1034}/braindecode/models/__init__.py

@@ -21,7 +21,7 @@ from .eegitnet import EEGITNet
 from .eegminer import EEGMiner
 from .eegnet import EEGNet, EEGNetv4
 from .eegnex import EEGNeX
-from .eegpt import EEGPT
+from .eegpt import EEGPT, InterpolatedEEGPT
 from .eegsimpleconv import EEGSimpleConv
 from .eegsym import EEGSym
 from .eegtcnet import EEGTCNet
@@ -85,6 +85,7 @@ __all__ = [
     "BrainModule",
     "EEGConformer",
     "EEGPT",
+    "InterpolatedEEGPT",
     "EEGInceptionERP",
     "EEGInceptionMI",
     "EEGITNet",

{braindecode-1.6.0.dev1028 → braindecode-1.6.1.dev1034}/braindecode/models/eegpt.py

@@ -1435,3 +1435,32 @@ class _EEGTransformer(nn.Module):
         )
 
         return x
+
+
+# -----------------------------------------------------------------------------
+# InterpolatedEEGPT — experimental channel-interpolation variant of EEGPT
+# -----------------------------------------------------------------------------
+# A :func:`~braindecode.models.interpolated.InterpolatedModel` wrapper around
+# :class:`EEGPT` whose target channel set is the canonical ``EEGPT_CHANNELS``.
+# Accepts arbitrary user ``chs_info``; projects to the canonical channels via an
+# MNE-backed (frozen by default) interpolation matrix.
+
+from mne.channels import make_standard_montage  # noqa: E402
+
+from braindecode.models.interpolated import InterpolatedModel  # noqa: E402
+
+montage = make_standard_montage("standard_1020")
+ch_pos = {
+    ch.upper(): (ch, loc) for ch, loc in montage.get_positions()["ch_pos"].items()
+}
+_EEGPT_TARGET_CHS_INFO = [
+    {
+        "ch_name": ch_pos[ch.upper()][0],
+        "kind": "eeg",
+        "loc": ch_pos[ch.upper()][1],
+    }
+    for ch in EEGPT_CHANNELS
+]
+InterpolatedEEGPT = InterpolatedModel(
+    EEGPT, _EEGPT_TARGET_CHS_INFO, name="InterpolatedEEGPT"
+)

{braindecode-1.6.0.dev1028 → braindecode-1.6.1.dev1034}/braindecode/models/summary.csv

@@ -53,5 +53,6 @@ REVE,General,Classification,200,"n_outputs, n_times, n_chans",69481476,"REVE(n_t
 DGCNN,Emotion Recognition,Classification,128,"n_chans, n_outputs, n_times, chs_info",1037385,"DGCNN(n_chans=62, n_outputs=4, n_times=128)","Graph Neural Network,Channel"
 InterpolatedBENDR,General,"Classification,Embedding",250,"chs_info, n_outputs, n_times",157143101,"InterpolatedBENDR(chs_info=<user>, n_outputs=4, n_times=1000)","Foundation Model,Convolution,Channel"
 InterpolatedBIOT,"Sleep Staging, Epilepsy",Classification,200,"chs_info, n_outputs, n_times, sfreq",3184101,"InterpolatedBIOT(chs_info=<user>, n_outputs=5, sfreq=200, n_times=6000)","Foundation Model,Channel"
+InterpolatedEEGPT,General,Classification,256,"chs_info, n_outputs, n_times",25323137,"InterpolatedEEGPT(chs_info=<user>, n_outputs=4, n_times=1024)","Foundation Model,Attention/Transformer,Channel"
 InterpolatedLaBraM,General,"Classification, Embedding",200,"chs_info, n_outputs, n_times",5866196,"InterpolatedLaBraM(chs_info=<user>, n_outputs=4, n_times=1000)","Convolution,Foundation Model,Channel"
 InterpolatedSignalJEPA,"Motor Imagery, ERP, SSVEP",Embedding,128,"chs_info",3456882,"InterpolatedSignalJEPA(chs_info=<user>)","Convolution,Channel,Foundation Model"

{braindecode-1.6.0.dev1028 → braindecode-1.6.1.dev1034}/braindecode/models/util.py

@@ -309,6 +309,11 @@ models_mandatory_parameters: list[
     ("EEGITNet", ["n_chans", "n_outputs", "n_times"], None),
     ("EEGNet", ["n_chans", "n_outputs", "n_times"], None),
     ("EEGPT", ["n_chans", "n_outputs", "n_times", "chs_info"], None),
+    (
+        "InterpolatedEEGPT",
+        ["chs_info", "n_outputs", "n_times"],
+        {"chs_info": _chs_info_4ch},  # MNE interpolation needs >=4 channels
+    ),
     ("ShallowFBCSPNet", ["n_chans", "n_outputs", "n_times"], None),
     (
         "SleepStagerBlanco2020",

braindecode-1.6.1.dev1034/braindecode/version.py

@@ -0,0 +1 @@
+__version__ = "1.6.1.dev1034"

{braindecode-1.6.0.dev1028 → braindecode-1.6.1.dev1034/braindecode.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: braindecode
-Version: 1.6.0.dev1028
+Version: 1.6.1.dev1034
 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.6.0.dev1028 → braindecode-1.6.1.dev1034}/docs/api.rst

@@ -160,6 +160,7 @@ interface for all EEG models and can derive variable names when needed.
      IFNet
      InterpolatedBENDR
      InterpolatedBIOT
+     InterpolatedEEGPT
      InterpolatedLaBraM
      InterpolatedModel
      InterpolatedSignalJEPA

{braindecode-1.6.0.dev1028 → braindecode-1.6.1.dev1034}/docs/whats_new.rst

@@ -22,12 +22,21 @@
 .. _current:
 
 
-Current 1.6.0 (GitHub)
+Current 1.6.1 (GitHub)
 ===============================
 
 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.
+- Add :class:`braindecode.models.InterpolatedEEGPT`, a channel-interpolation
+  variant of :class:`braindecode.models.EEGPT` built with
+  :func:`~braindecode.models.interpolated.InterpolatedModel`.
+  By `Pierre Guetschel`_.
+
 API and behavior changes
 ========================
 
@@ -50,10 +59,18 @@ Bug fixes
   use running statistics when ``batch_size == 1`` in train mode.
   By `Bruno Aristimunha`_.
 
+- Fix incorrect import path in CONTRIBUTING.md by `Yiheng Li`_
+
 Code health
 ============
 
-- None yet
+- 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)
@@ -1224,3 +1241,5 @@ Authors
 .. _Sarthak Tayal: https://github.com/tayal-sarthak
 .. _Vandit Shah: https://github.com/ShahVandit
 .. _Léo Burgund: https://github.com/leob000
+.. _Adam Mounir: https://github.com/adammounir
+.. _Yiheng Li: https://github.com/YihengLi-1

braindecode-1.6.0.dev1028/braindecode/version.py

@@ -1 +0,0 @@
-__version__ = "1.6.0.dev1028"