Braindecode 1.5.0.dev1022__tar.gz → 1.5.0.dev174273210__tar.gz

{braindecode-1.5.0.dev1022 → braindecode-1.5.0.dev174273210}/NOTICE.txt

@@ -11,14 +11,9 @@ All files within the `src/` directory are licensed under the BSD-3-Clause Licens
 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)

{braindecode-1.5.0.dev1022/braindecode.egg-info → braindecode-1.5.0.dev174273210}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: braindecode
-Version: 1.5.0.dev1022
+Version: 1.5.0.dev174273210
 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>
@@ -43,8 +43,6 @@ Requires-Dist: eegprep[eeglabio]>=0.2.23; extra == "eegprep"
 Provides-Extra: hub
 Requires-Dist: huggingface_hub[torch]>=0.20.0; extra == "hub"
 Requires-Dist: zarr>=3.0; extra == "hub"
-Provides-Extra: viz
-Requires-Dist: captum>=0.7.0; extra == "viz"
 Provides-Extra: tests
 Requires-Dist: pytest; extra == "tests"
 Requires-Dist: pytest-cov; extra == "tests"
@@ -63,7 +61,6 @@ Requires-Dist: sphinx-autodoc-typehints; extra == "docs"
 Requires-Dist: sphinx-autobuild; extra == "docs"
 Requires-Dist: sphinxcontrib-bibtex; extra == "docs"
 Requires-Dist: sphinx_sitemap; extra == "docs"
-Requires-Dist: sphinxext-opengraph; extra == "docs"
 Requires-Dist: pydata_sphinx_theme; extra == "docs"
 Requires-Dist: numpydoc; extra == "docs"
 Requires-Dist: memory_profiler; extra == "docs"
@@ -73,7 +70,7 @@ Requires-Dist: sphinx_design; extra == "docs"
 Requires-Dist: lightning; extra == "docs"
 Requires-Dist: seaborn; extra == "docs"
 Requires-Dist: pre-commit; extra == "docs"
-Requires-Dist: openneuro-py>=2026.4.0; extra == "docs"
+Requires-Dist: openneuro-py; extra == "docs"
 Requires-Dist: plotly; extra == "docs"
 Requires-Dist: shap; extra == "docs"
 Requires-Dist: nbformat; extra == "docs"
@@ -85,7 +82,6 @@ Requires-Dist: braindecode[docs]; extra == "all"
 Requires-Dist: braindecode[hub]; extra == "all"
 Requires-Dist: braindecode[eegprep]; extra == "all"
 Requires-Dist: braindecode[typing]; extra == "all"
-Requires-Dist: braindecode[viz]; extra == "all"
 Dynamic: license-file
 
 .. image:: https://zenodo.org/badge/DOI/10.5281/zenodo.17699192.svg

{braindecode-1.5.0.dev1022 → braindecode-1.5.0.dev174273210}/braindecode/augmentation/__init__.py

@@ -4,7 +4,6 @@ from . import functional
 from .base import AugmentedDataLoader, Compose, IdentityTransform, Transform
 from .transforms import (
     AmplitudeScale,
-    BandRotation,
     BandstopFilter,
     ChannelsDropout,
     ChannelsReref,
@@ -48,7 +47,6 @@ __all__ = [
     "SegmentationReconstruction",
     "MaskEncoding",
     "AmplitudeScale",
-    "BandRotation",
     "ChannelsReref",
     "functional",
 ]

{braindecode-1.5.0.dev1022 → braindecode-1.5.0.dev174273210}/braindecode/augmentation/functional.py

@@ -1298,130 +1298,3 @@ def amplitude_scale(
     X = s * X
 
     return X, y
-
-
-def band_rotation(
-    X: torch.Tensor,
-    y: torch.Tensor,
-    num_bands: int = 2,
-    electrodes_per_band: int = 16,
-    band_offsets: tuple[int, ...] = (-1, 0, 1),
-    max_temporal_jitter: int = 0,
-    circular_jitter: bool = True,
-    random_state: int | np.random.RandomState | None = None,
-) -> tuple[torch.Tensor, torch.Tensor]:
-    """Per-band electrode rotation + inter-band temporal jitter.
-
-    Models small wristband rotation between sessions and relative timing
-    noise between two arms.  Introduced in [Sivakumar2024]_ for the
-    emg2qwerty CTC keystroke decoding task: each electrode band gets its
-    own circular roll along the channel axis (``Uniform(band_offsets)``
-    positions), and band 1 also gets a sample-level temporal shift
-    (``Uniform(-max_temporal_jitter, +max_temporal_jitter)``) along the
-    time axis.
-
-    Channel layout assumes ``(B, num_bands * electrodes_per_band, T)`` with
-    bands contiguous along the channel axis.  Same offset / shift is
-    applied to every sample in the batch (one set of parameters per call).
-
-    Parameters
-    ----------
-    X : torch.Tensor
-        EMG input batch of shape ``(B, C, T)`` with
-        ``C == num_bands * electrodes_per_band``.
-    y : torch.Tensor
-        Labels (returned unchanged).
-    num_bands : int, optional
-        Number of electrode bands (e.g. ``2`` for left + right wristband).
-        Must be ``>= 1``.  Defaults to 2.
-    electrodes_per_band : int, optional
-        Electrodes per band (e.g. ``16``).  Must be ``>= 1``.  Defaults
-        to 16.
-    band_offsets : tuple of int, optional
-        Per-band roll values to sample from uniformly.  ``(-1, 0, 1)``
-        covers ±1-electrode misalignment.  Must be non-empty.  Defaults
-        to ``(-1, 0, 1)``.
-    max_temporal_jitter : int, optional
-        Max ±-sample temporal shift applied to band 1 only when
-        ``num_bands >= 2``.  Defaults to 0 (disabled).  Must be ``>= 0``.
-    circular_jitter : bool, optional
-        If True (the default, paper-faithful), the temporal jitter is a
-        circular ``torch.roll`` — samples shifted off one edge wrap to
-        the other.  If False, the gap left by the shift is zero-padded
-        and the shifted-off samples are dropped, avoiding wrap-around
-        discontinuity at the cost of a small zeroed margin.  Has no
-        effect when ``max_temporal_jitter == 0``.
-    random_state : int | numpy.random.RandomState, optional
-        Seed / generator for sampling rotation + jitter values.
-
-    Returns
-    -------
-    torch.Tensor
-        Transformed inputs.
-    torch.Tensor
-        Labels (unchanged).
-
-    References
-    ----------
-    .. [Sivakumar2024] Sivakumar, V., Seely, J., Du, A., Bittner, S. R.,
-       Berenzweig, A., Bolarinwa, A., Gramfort, A., & Mandel, M. I. (2024).
-       "emg2qwerty: A Large Dataset with Baselines for Touch Typing using
-       Surface Electromyography." *NeurIPS Datasets and Benchmarks Track*.
-    """
-    if num_bands < 1:
-        raise ValueError(f"num_bands must be >= 1, got {num_bands}")
-    if electrodes_per_band < 1:
-        raise ValueError(f"electrodes_per_band must be >= 1, got {electrodes_per_band}")
-    # Normalise to a tuple before truth-testing so callers can pass any
-    # sequence-like (incl. ``np.ndarray``) without hitting numpy's
-    # ambiguous-truth-value error on ``if not band_offsets``.
-    band_offsets = tuple(band_offsets)
-    if not band_offsets:
-        raise ValueError("band_offsets must be non-empty")
-    if not all(isinstance(o, (int, np.integer)) for o in band_offsets):
-        raise ValueError(f"band_offsets must contain integers, got {band_offsets!r}")
-    if max_temporal_jitter < 0:
-        raise ValueError(f"max_temporal_jitter must be >= 0, got {max_temporal_jitter}")
-    expected_channels = num_bands * electrodes_per_band
-    if X.shape[1] != expected_channels:
-        raise ValueError(
-            f"X.shape[1]={X.shape[1]} != num_bands * electrodes_per_band="
-            f"{expected_channels}"
-        )
-
-    rng = check_random_state(random_state)
-    band_offsets_arr = np.asarray(band_offsets)
-    out = X.clone()
-
-    # Per-band channel-axis rolls.  A vectorized ``torch.gather`` was
-    # benchmarked and is ~16 % slower for the typical ``num_bands == 2``
-    # case on CPU (the index tensor is larger than what two contiguous
-    # rolls touch); the gather only wins past ``num_bands >= 8``.
-    for b in range(num_bands):
-        offset = int(rng.choice(band_offsets_arr))
-        if offset:
-            sl = slice(b * electrodes_per_band, (b + 1) * electrodes_per_band)
-            out[:, sl, :] = torch.roll(out[:, sl, :], offset, dims=1)
-
-    # Inter-band temporal jitter — paper recipe applies it to band 1 only.
-    if max_temporal_jitter > 0 and num_bands >= 2:
-        shift = int(rng.randint(-max_temporal_jitter, max_temporal_jitter + 1))
-        if shift:
-            sl = slice(electrodes_per_band, 2 * electrodes_per_band)
-            band1 = out[:, sl, :]
-            if circular_jitter:
-                # Paper-faithful circular shift; wraps end-of-window
-                # samples to the start (and vice versa).
-                out[:, sl, :] = torch.roll(band1, shift, dims=2)
-            else:
-                # Crop-and-pad shift: drop samples that fall off one end,
-                # zero-pad the gap on the other.  Avoids the wrap-around
-                # discontinuity at the cost of a ``|shift|``-sample margin.
-                shifted = torch.zeros_like(band1)
-                if shift > 0:
-                    shifted[:, :, shift:] = band1[:, :, :-shift]
-                else:  # shift < 0
-                    shifted[:, :, :shift] = band1[:, :, -shift:]
-                out[:, sl, :] = shifted
-
-    return out, y

{braindecode-1.5.0.dev1022 → braindecode-1.5.0.dev174273210}/braindecode/augmentation/transforms.py

@@ -16,7 +16,6 @@ from mne.channels import make_standard_montage
 from .base import Transform
 from .functional import (
     amplitude_scale,
-    band_rotation,
     bandstop_filter,
     channels_dropout,
     channels_permute,
@@ -1357,99 +1356,3 @@ class AmplitudeScale(Transform):
     def get_augmentation_params(self, *batch):
         """Return transform parameters."""
         return {"random_state": self.rng, "scale": self.scale}
-
-
-class BandRotation(Transform):
-    """Per-band electrode rotation + inter-band temporal jitter.
-
-    Models small wristband rotation between sessions and relative timing
-    noise between two arms.  Introduced in [Sivakumar2024]_ for the
-    emg2qwerty surface-EMG keystroke decoding task: the channel axis is
-    laid out as ``(B, num_bands * electrodes_per_band, T)`` with bands
-    contiguous, each band gets a uniform circular roll along the channel
-    axis, and when ``num_bands >= 2``, band 1 also gets a sample-level
-    temporal shift.  The same offset / shift is applied to every sample
-    in a transformed sub-batch (one set of parameters per call).
-
-    Parameters
-    ----------
-    probability : float
-        Float setting the probability of applying the operation.
-    num_bands : int, optional
-        Number of electrode bands (e.g. ``2`` for left + right wristband).
-        Must be ``>= 1``.  Defaults to 2.
-    electrodes_per_band : int, optional
-        Electrodes per band (e.g. ``16``).  Must be ``>= 1``.  Defaults
-        to 16.
-    band_offsets : tuple of int, optional
-        Per-band roll values to sample from uniformly.  ``(-1, 0, 1)``
-        covers ±1-electrode misalignment.  Must be non-empty.  Defaults
-        to ``(-1, 0, 1)``.
-    max_temporal_jitter : int, optional
-        Max ±-sample temporal shift applied to band 1.  Defaults to 0
-        (jitter disabled).  Must be ``>= 0``.  The emg2qwerty paper uses
-        120 samples (60 ms at 2 kHz).
-    circular_jitter : bool, optional
-        If True (default, paper-faithful) the jitter is a circular roll;
-        if False the gap left by the shift is zero-padded.  See
-        :func:`band_rotation`.
-    random_state : int | numpy.random.RandomState, optional
-        Seed for the rotation / jitter sampler.  Defaults to None.
-
-    References
-    ----------
-    .. [Sivakumar2024] Sivakumar, V., Seely, J., Du, A., Bittner, S. R.,
-       Berenzweig, A., Bolarinwa, A., Gramfort, A., & Mandel, M. I. (2024).
-       "emg2qwerty: A Large Dataset with Baselines for Touch Typing using
-       Surface Electromyography." *NeurIPS Datasets and Benchmarks Track*.
-    """
-
-    operation = staticmethod(band_rotation)  # type: ignore[assignment]
-
-    def __init__(
-        self,
-        probability,
-        num_bands=2,
-        electrodes_per_band=16,
-        band_offsets=(-1, 0, 1),
-        max_temporal_jitter=0,
-        circular_jitter=True,
-        random_state=None,
-    ):
-        super().__init__(probability=probability, random_state=random_state)
-        # Up-front parameter validation; the underlying ``band_rotation``
-        # also re-checks at call time, but raising here surfaces config
-        # mistakes when the Transform is built rather than on the first
-        # batch.
-        if num_bands < 1:
-            raise ValueError(f"num_bands must be >= 1, got {num_bands}")
-        if electrodes_per_band < 1:
-            raise ValueError(
-                f"electrodes_per_band must be >= 1, got {electrodes_per_band}"
-            )
-        band_offsets = tuple(band_offsets)
-        if not band_offsets:
-            raise ValueError("band_offsets must be non-empty")
-        if not all(isinstance(o, (int, np.integer)) for o in band_offsets):
-            raise ValueError(
-                f"band_offsets must contain integers, got {band_offsets!r}"
-            )
-        if max_temporal_jitter < 0:
-            raise ValueError(
-                f"max_temporal_jitter must be >= 0, got {max_temporal_jitter}"
-            )
-        self.num_bands = num_bands
-        self.electrodes_per_band = electrodes_per_band
-        self.band_offsets = band_offsets
-        self.max_temporal_jitter = max_temporal_jitter
-        self.circular_jitter = circular_jitter
-
-    def get_augmentation_params(self, *batch):
-        return {
-            "num_bands": self.num_bands,
-            "electrodes_per_band": self.electrodes_per_band,
-            "band_offsets": self.band_offsets,
-            "max_temporal_jitter": self.max_temporal_jitter,
-            "circular_jitter": self.circular_jitter,
-            "random_state": self.rng,
-        }

{braindecode-1.5.0.dev1022 → braindecode-1.5.0.dev174273210}/braindecode/datasets/base.py

@@ -18,7 +18,7 @@ import shutil
 import warnings
 from abc import abstractmethod
 from collections import Counter
-from collections.abc import Callable, Hashable
+from collections.abc import Callable
 from glob import glob
 from pathlib import Path
 from typing import Any, Generic, Iterable, no_type_check
@@ -1340,124 +1340,6 @@ class BaseConcatDataset(ConcatDataset, HubDatasetMixin, Generic[T]):
             raise TypeError("target_transform must be a callable.")
         self._target_transform = fn
 
-    def set_target(self, column: Hashable) -> "BaseConcatDataset":
-        """Use ``column`` as the target ``y`` for every subdataset.
-
-        Dispatches on the subdataset type:
-
-        * For :class:`WindowsDataset` / :class:`EEGWindowsDataset`,
-          ``column`` is looked up in per-window ``metadata`` first, then in
-          the per-record ``description`` (broadcast to every window). The
-          resolved values overwrite ``ds.metadata['target']`` and ``ds.y``.
-          For :class:`WindowsDataset`, the underlying ``ds.windows.metadata``
-          is kept in sync so ``get_metadata()`` and the repr reflect the
-          new target.
-        * For :class:`RawDataset`, ``column`` must exist on the
-          ``description``. ``ds.target_name`` is set to ``column`` so
-          ``__getitem__`` reads ``description[column]`` as ``y`` on every
-          access — no rebuild needed.
-
-        Parameters
-        ----------
-        column : Hashable
-            Name of a metadata column or description field (BIDS entity,
-            participants.tsv extra, ...). Typically a string, but any
-            hashable that pandas accepts as a column label is allowed.
-
-        Returns
-        -------
-        self : BaseConcatDataset
-
-        Raises
-        ------
-        TypeError
-            If any subdataset is not a :class:`WindowsDataset`,
-            :class:`EEGWindowsDataset`, or :class:`RawDataset`, or if a
-            windowed subdataset has lazy (non-DataFrame) metadata.
-        ValueError
-            If ``column`` is not present on a subdataset's metadata or
-            description, or if a windowed subdataset has
-            ``targets_from='channels'`` (which would make this a silent
-            no-op since ``__getitem__`` reads y from misc channels, not
-            from ``metadata['target']``).
-        """
-        for i, ds in enumerate(self.datasets):
-            if isinstance(ds, (WindowsDataset, EEGWindowsDataset)):
-                if not isinstance(ds.metadata, pd.DataFrame):
-                    # _LazyDataFrame (lazy_metadata=True) does not implement
-                    # .copy()/__setitem__, so the in-place write below would
-                    # raise AttributeError. Surface the precondition cleanly.
-                    raise TypeError(
-                        "set_target requires a materialized metadata "
-                        f"DataFrame; datasets[{i}].metadata is "
-                        f"{type(ds.metadata).__name__}. Re-window with "
-                        "lazy_metadata=False to use set_target."
-                    )
-                if getattr(ds, "targets_from", "metadata") != "metadata":
-                    # __getitem__ would read y from misc channels; writing
-                    # metadata['target']/ds.y would be a silent no-op.
-                    raise ValueError(
-                        f"datasets[{i}] has targets_from="
-                        f"{ds.targets_from!r}; set_target only applies when "
-                        "targets_from='metadata' (otherwise __getitem__ "
-                        "derives y from misc channels and would ignore the "
-                        "rewritten target column)."
-                    )
-                n = len(ds)
-                md = ds.metadata
-                if column in md.columns:
-                    values = md[column].iloc[:n].to_list()
-                elif (
-                    isinstance(ds.description, pd.Series)
-                    and column in ds.description.index
-                ):
-                    values = [ds.description[column]] * n
-                else:
-                    desc_keys = (
-                        list(ds.description.index)
-                        if isinstance(ds.description, pd.Series)
-                        else []
-                    )
-                    raise ValueError(
-                        f"Column {column!r} not found on datasets[{i}]: "
-                        f"metadata cols={list(md.columns)}, "
-                        f"description keys={desc_keys}."
-                    )
-                # In-place write so the WindowsDataset's metadata and the
-                # underlying mne.Epochs.metadata (which start as the same
-                # object reference) both reflect the new target. Defensive
-                # second write covers the case where they got de-aliased
-                # earlier by a caller-side reassignment.
-                md["target"] = values
-                windows_obj = getattr(ds, "_windows", None)
-                if windows_obj is not None and windows_obj.metadata is not md:
-                    windows_obj.metadata["target"] = values
-                # values is already a fresh list (Series.to_list() / [x] * n);
-                # no defensive copy needed — pandas keeps its own representation
-                # for md["target"] so mutating ds.y won't reach back into it.
-                ds.y = values
-            elif isinstance(ds, RawDataset):
-                if (
-                    not isinstance(ds.description, pd.Series)
-                    or column not in ds.description.index
-                ):
-                    desc_keys = (
-                        list(ds.description.index)
-                        if isinstance(ds.description, pd.Series)
-                        else []
-                    )
-                    raise ValueError(
-                        f"Column {column!r} not found on datasets[{i}] "
-                        f"description (keys={desc_keys})."
-                    )
-                ds.target_name = column
-            else:
-                raise TypeError(
-                    "set_target requires WindowsDataset, EEGWindowsDataset, "
-                    f"or RawDataset; datasets[{i}] is {type(ds).__name__}."
-                )
-        return self
-
     def _outdated_save(self, path, overwrite=False):
         """This is a copy of the old saving function, that had inconsistent.
 

{braindecode-1.5.0.dev1022 → braindecode-1.5.0.dev174273210}/braindecode/datasets/sleep_physionet.py

@@ -6,12 +6,10 @@
 from __future__ import annotations
 
 import os
-import time
 
 import mne
 import numpy as np
 import pandas as pd
-import requests
 from mne.datasets.sleep_physionet.age import fetch_data
 
 from .base import BaseConcatDataset, RawDataset
@@ -61,19 +59,7 @@ class SleepPhysionet(BaseConcatDataset):
         if recording_ids is None:
             recording_ids = [1, 2]
 
-        max_attempts = 5
-        retry_delay_s = 10
-
-        for attempt in range(max_attempts):
-            try:
-                paths = fetch_data(
-                    subject_ids, recording=recording_ids, on_missing="warn"
-                )
-                break
-            except requests.exceptions.RequestException:
-                if attempt == max_attempts - 1:
-                    raise
-                time.sleep(retry_delay_s)
+        paths = fetch_data(subject_ids, recording=recording_ids, on_missing="warn")
 
         all_base_ds = list()
         for p in paths:

{braindecode-1.5.0.dev1022 → braindecode-1.5.0.dev174273210}/braindecode/models/__init__.py

@@ -4,8 +4,8 @@ from .atcnet import ATCNet
 from .attentionbasenet import AttentionBaseNet
 from .attn_sleep import AttnSleep
 from .base import EEGModuleMixin
-from .bendr import BENDR, InterpolatedBENDR
-from .biot import BIOT, InterpolatedBIOT
+from .bendr import BENDR
+from .biot import BIOT
 from .brainmodule import BrainModule
 from .cbramod import CBraMod
 from .codebrain import CodeBrain
@@ -25,24 +25,20 @@ from .eegpt import EEGPT
 from .eegsimpleconv import EEGSimpleConv
 from .eegsym import EEGSym
 from .eegtcnet import EEGTCNet
-from .emg2qwerty import EMG2QwertyNet
 from .fbcnet import FBCNet
 from .fblightconvnet import FBLightConvNet
 from .fbmsnet import FBMSNet
 from .hybrid import HybridNet
 from .ifnet import IFNet
-from .interpolated import InterpolatedModel
-from .labram import InterpolatedLaBraM, Labram
+from .labram import Labram
 from .luna import LUNA
 from .medformer import MEDFormer
-from .meta_neuromotor import MetaNeuromotorHand
 from .msvtnet import MSVTNet
 from .patchedtransformer import PBT
 from .reve import REVE
 from .sccnet import SCCNet
 from .shallow_fbcsp import ShallowFBCSPNet
 from .signal_jepa import (
-    InterpolatedSignalJEPA,
     SignalJEPA,
     SignalJEPA_Contextual,
     SignalJEPA_PostLocal,
@@ -96,18 +92,11 @@ __all__ = [
     "EEGSimpleConv",
     "EEGTCNet",
     "DGCNN",
-    "EMG2QwertyNet",
     "FBCNet",
     "FBLightConvNet",
     "FBMSNet",
-    "MetaNeuromotorHand",
     "HybridNet",
     "IFNet",
-    "InterpolatedBENDR",
-    "InterpolatedBIOT",
-    "InterpolatedLaBraM",
-    "InterpolatedModel",
-    "InterpolatedSignalJEPA",
     "Labram",
     "LUNA",
     "extract_channel_locations_from_chs_info",

{braindecode-1.5.0.dev1022 → braindecode-1.5.0.dev174273210}/braindecode/models/base.py

@@ -33,25 +33,13 @@ huggingface_hub = _soft_import(
 HAS_HF_HUB = huggingface_hub is not False
 
 
-_HF_INSTALL_HINT = (
-    "requires the `huggingface_hub` package. "
-    "Install with: pip install 'braindecode[hub]'"
-)
-
-
-class _BaseHubMixinStub:
-    @classmethod
-    def from_pretrained(cls, *args, **kwargs):
-        raise ImportError(f"{cls.__name__}.from_pretrained() {_HF_INSTALL_HINT}")
-
-    def push_to_hub(self, *args, **kwargs):
-        raise ImportError(f"{type(self).__name__}.push_to_hub() {_HF_INSTALL_HINT}")
+class _BaseHubMixin:
+    pass
 
 
 # Define base class for hub mixin
-_BaseHubMixin: Type = (
-    huggingface_hub.PyTorchModelHubMixin if HAS_HF_HUB else _BaseHubMixinStub
-)
+if HAS_HF_HUB:
+    _BaseHubMixin: Type = huggingface_hub.PyTorchModelHubMixin  # type: ignore
 
 
 def deprecated_args(obj, *old_new_args):