Braindecode 1.3.0.dev175887834__tar.gz → 1.3.0.dev176089716__tar.gz

{braindecode-1.3.0.dev175887834/braindecode.egg-info → braindecode-1.3.0.dev176089716}/PKG-INFO

@@ -1,8 +1,8 @@
 Metadata-Version: 2.4
 Name: braindecode
-Version: 1.3.0.dev175887834
+Version: 1.3.0.dev176089716
 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>
+Author-email: Robin Tibor Schirrmeister <robintibor@gmail.com>
 Maintainer-email: Alexandre Gramfort <agramfort@meta.com>, Bruno Aristimunha Pinto <b.aristimunha@gmail.com>, Robin Tibor Schirrmeister <robintibor@gmail.com>
 License: BSD-3-Clause
 Project-URL: homepage, https://braindecode.org
@@ -14,47 +14,42 @@ Classifier: Intended Audience :: Developers
 Classifier: Intended Audience :: Science/Research
 Classifier: Topic :: Software Development :: Build Tools
 Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
-Classifier: Programming Language :: Python :: 3.13
-Classifier: Programming Language :: Python :: 3.14
-Requires-Python: >=3.11
+Requires-Python: >=3.10
 Description-Content-Type: text/x-rst
 License-File: LICENSE.txt
 License-File: NOTICE.txt
-Requires-Dist: mne>=1.11.0
+Requires-Dist: mne>=1.10.0
 Requires-Dist: mne_bids>=0.16
 Requires-Dist: numpy
 Requires-Dist: pandas
 Requires-Dist: scipy
 Requires-Dist: matplotlib
 Requires-Dist: h5py
-Requires-Dist: skorch>=1.3.0
-Requires-Dist: torch>=2.0
-Requires-Dist: torchaudio>=2.0
+Requires-Dist: skorch>=1.2.0
+Requires-Dist: torch<3.0,>=2.0
+Requires-Dist: torchaudio<3.0,>=2.0
 Requires-Dist: einops
 Requires-Dist: joblib
-Requires-Dist: torchinfo
+Requires-Dist: torchinfo~=1.8
 Requires-Dist: wfdb
+Requires-Dist: h5py
 Requires-Dist: linear_attention_transformer
 Requires-Dist: docstring_inheritance
-Requires-Dist: rotary_embedding_torch
 Provides-Extra: moabb
-Requires-Dist: moabb>=1.4.2; extra == "moabb"
+Requires-Dist: moabb>=1.2.0; extra == "moabb"
 Provides-Extra: eegprep
-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"
+Requires-Dist: eegprep[eeglabio]>=0.1.1; extra == "eegprep"
+Provides-Extra: hug
+Requires-Dist: huggingface_hub[torch]>=0.20.0; extra == "hug"
 Provides-Extra: tests
 Requires-Dist: pytest; extra == "tests"
 Requires-Dist: pytest-cov; extra == "tests"
 Requires-Dist: codecov; extra == "tests"
 Requires-Dist: pytest_cases; extra == "tests"
 Requires-Dist: mypy; extra == "tests"
-Provides-Extra: typing
-Requires-Dist: exca==0.4; extra == "typing"
-Requires-Dist: numpydantic>=1.7; extra == "typing"
 Provides-Extra: docs
 Requires-Dist: sphinx_gallery; extra == "docs"
 Requires-Dist: sphinx_rtd_theme; extra == "docs"
@@ -73,21 +68,10 @@ Requires-Dist: seaborn; extra == "docs"
 Requires-Dist: pre-commit; extra == "docs"
 Requires-Dist: openneuro-py; extra == "docs"
 Requires-Dist: plotly; extra == "docs"
-Requires-Dist: shap; extra == "docs"
-Requires-Dist: nbformat; extra == "docs"
 Provides-Extra: all
-Requires-Dist: braindecode[moabb]; extra == "all"
-Requires-Dist: braindecode[tests]; extra == "all"
-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[docs,eegprep,hug,moabb,tests]; extra == "all"
 Dynamic: license-file
 
-.. image:: https://zenodo.org/badge/232335424.svg
-   :target: https://doi.org/10.5281/zenodo.8214376
-   :alt: DOI
-
 .. image:: https://badges.gitter.im/braindecodechat/community.svg
    :alt: Join the chat at https://gitter.im/braindecodechat/community
    :target: https://gitter.im/braindecodechat/community?utm_source=badge&utm_medium=badge&utm_campaign=pr-badge&utm_content=badge
@@ -103,7 +87,6 @@ Dynamic: license-file
    :target: https://codecov.io/gh/braindecode/braindecode
    :alt: Code Coverage
 
-
 .. |Braindecode| image:: https://user-images.githubusercontent.com/42702466/177958779-b00628aa-9155-4c51-96d1-d8c345aff575.svg
 .. _braindecode: braindecode.org/
 
@@ -160,15 +143,7 @@ https://gitter.im/braindecodechat/community
 Citing
 ======
 
-If you use Braindecode in scientific work, please cite the software using the Zenodo DOI shown in the badge below:
-
-.. image:: https://zenodo.org/badge/232335424.svg
-   :target: https://doi.org/10.5281/zenodo.8214376
-   :alt: DOI
-
-Additionally, we highly encourage you to cite the article that originally introduced
-the Braindecode library and has served as a foundational reference for many works
-on deep learning with EEG recordings. Please use the following reference:
+If you use this code in a scientific publication, please cite us as:
 
 .. code-block:: bibtex
 

{braindecode-1.3.0.dev175887834 → braindecode-1.3.0.dev176089716}/README.rst

@@ -1,7 +1,3 @@
-.. image:: https://zenodo.org/badge/232335424.svg
-   :target: https://doi.org/10.5281/zenodo.8214376
-   :alt: DOI
-
 .. image:: https://badges.gitter.im/braindecodechat/community.svg
    :alt: Join the chat at https://gitter.im/braindecodechat/community
    :target: https://gitter.im/braindecodechat/community?utm_source=badge&utm_medium=badge&utm_campaign=pr-badge&utm_content=badge
@@ -17,7 +13,6 @@
    :target: https://codecov.io/gh/braindecode/braindecode
    :alt: Code Coverage
 
-
 .. |Braindecode| image:: https://user-images.githubusercontent.com/42702466/177958779-b00628aa-9155-4c51-96d1-d8c345aff575.svg
 .. _braindecode: braindecode.org/
 
@@ -74,15 +69,7 @@ https://gitter.im/braindecodechat/community
 Citing
 ======
 
-If you use Braindecode in scientific work, please cite the software using the Zenodo DOI shown in the badge below:
-
-.. image:: https://zenodo.org/badge/232335424.svg
-   :target: https://doi.org/10.5281/zenodo.8214376
-   :alt: DOI
-
-Additionally, we highly encourage you to cite the article that originally introduced
-the Braindecode library and has served as a foundational reference for many works
-on deep learning with EEG recordings. Please use the following reference:
+If you use this code in a scientific publication, please cite us as:
 
 .. code-block:: bibtex
 

{braindecode-1.3.0.dev175887834 → braindecode-1.3.0.dev176089716}/braindecode/augmentation/__init__.py

@@ -5,10 +5,8 @@ Utilities for data augmentation.
 from . import functional
 from .base import AugmentedDataLoader, Compose, IdentityTransform, Transform
 from .transforms import (
-    AmplitudeScale,
     BandstopFilter,
     ChannelsDropout,
-    ChannelsReref,
     ChannelsShuffle,
     ChannelsSymmetry,
     FrequencyShift,
@@ -48,7 +46,5 @@ __all__ = [
     "Mixup",
     "SegmentationReconstruction",
     "MaskEncoding",
-    "AmplitudeScale",
-    "ChannelsReref",
     "functional",
 ]

{braindecode-1.3.0.dev175887834 → braindecode-1.3.0.dev176089716}/braindecode/augmentation/functional.py

@@ -1,7 +1,6 @@
 # Authors: Cédric Rommel <cedric.rommel@inria.fr>
 #          Alexandre Gramfort <alexandre.gramfort@inria.fr>
 #          Gustavo Rodrigues <gustavenrique01@gmail.com>
-#          Bruna Lopes <brunajaflopes@gmail.com>
 #
 # License: BSD (3-clause)
 
@@ -1195,103 +1194,3 @@ def mask_encoding(
     X[mask] = 0
 
     return X, y  # Return the masked tensor and labels
-
-
-def channels_rereference(
-    X: torch.Tensor,
-    y: torch.Tensor,
-    random_state: int | np.random.RandomState | None = None,
-) -> tuple[torch.Tensor, torch.Tensor]:
-    """Randomly re-reference channels in EEG data matrix.
-
-    Part of the augmentations proposed in [1]_
-
-    Parameters
-    ----------
-    X : torch.Tensor
-        EEG input example or batch.
-    y : torch.Tensor
-        EEG labels for the example or batch.
-    random_state: int | numpy.random.Generator, optional
-        Seed to be used to instantiate numpy random number generator instance.
-        Defaults to None.
-
-    Returns
-    -------
-    torch.Tensor
-        Transformed inputs.
-    torch.Tensor
-        Transformed labels.
-
-    References
-    ----------
-    .. [1] Mohsenvand, M.N., Izadi, M.R. &amp; Maes, P.. (2020). Contrastive
-        Representation Learning for Electroencephalogram Classification. Proceedings
-        of the Machine Learning for Health NeurIPS Workshop, in Proceedings of Machine
-        Learning Research 136:238-253
-
-    """
-
-    rng = check_random_state(random_state)
-    batch_size, n_channels, _ = X.shape
-
-    ch = rng.randint(0, n_channels, size=batch_size)
-
-    X_ch = X[torch.arange(batch_size), ch, :]
-    X = X - X_ch.unsqueeze(1)
-    X[torch.arange(batch_size), ch, :] = -X_ch
-
-    return X, y
-
-
-def amplitude_scale(
-    X: torch.Tensor,
-    y: torch.Tensor,
-    scale: tuple,
-    random_state: int | np.random.RandomState | None = None,
-) -> tuple[torch.Tensor, torch.Tensor]:
-    """Rescale amplitude of each channel based on a random sampled scaling value.
-
-    Part of the augmentations proposed in [1]_
-
-    Parameters
-    ----------
-    X : torch.Tensor
-        EEG input example or batch.
-    y : torch.Tensor
-        EEG labels for the example or batch.
-    scale : tuple of floats
-        Interval from which ypu sample the scaling value
-    random_state: int | numpy.random.Generator, optional
-        Seed to be used to instantiate numpy random number generator instance.
-        Defaults to None.
-
-    Returns
-    -------
-    torch.Tensor
-        Transformed inputs.
-    torch.Tensor
-        Transformed labels.
-
-    References
-    ----------
-    .. [1] Mohsenvand, M.N., Izadi, M.R. &amp; Maes, P.. (2020). Contrastive
-        Representation Learning for Electroencephalogram Classification. Proceedings
-        of the Machine Learning for Health NeurIPS Workshop, in Proceedings of Machine
-        Learning Research 136:238-253
-
-    """
-
-    rng = torch.Generator()
-    rng.manual_seed(random_state)
-    batch_size, n_channels, _ = X.shape
-
-    # Parameter for scaling amplitude / channel / trial
-    l, h = scale
-    s = l + (h - l) * torch.rand(
-        batch_size, n_channels, 1, generator=rng, device=X.device, dtype=X.dtype
-    )
-
-    X = s * X
-
-    return X, y

{braindecode-1.3.0.dev175887834 → braindecode-1.3.0.dev176089716}/braindecode/augmentation/transforms.py

@@ -1,7 +1,6 @@
 # Authors: Cédric Rommel <cedric.rommel@inria.fr>
 #          Alexandre Gramfort <alexandre.gramfort@inria.fr>
 #          Gustavo Rodrigues <gustavenrique01@gmail.com>
-#          Bruna Lopes <brunajaflopes@gmail.com>
 #
 # License: BSD (3-clause)
 
@@ -14,11 +13,9 @@ from mne.channels import make_standard_montage
 
 from .base import Transform
 from .functional import (
-    amplitude_scale,
     bandstop_filter,
     channels_dropout,
     channels_permute,
-    channels_rereference,
     channels_shuffle,
     frequency_shift,
     ft_surrogate,
@@ -1274,74 +1271,3 @@ class MaskEncoding(Transform):
             "segment_length": segment_length,
             "n_segments": self.n_segments,
         }
-
-
-class ChannelsReref(Transform):
-    """Randomly re-reference channels in EEG data matrix.
-
-    Part of the augmentations proposed in [1]_
-
-    Parameters
-    ----------
-    probability: float
-        Float setting the probability of applying the operation.
-    random_state: int | numpy.random.Generator, optional
-        Seed to be used to instantiate numpy random number generator instance.
-        Used to decide whether or not to transform given the probability
-        argument, to sample which channels to shuffle and to carry the shuffle.
-        Defaults to None.
-
-    References
-    ----------
-    .. [1] Mohsenvand, M.N., Izadi, M.R. &amp; Maes, P.. (2020). Contrastive
-        Representation Learning for Electroencephalogram Classification. Proceedings
-        of the Machine Learning for Health NeurIPS Workshop, in Proceedings of Machine
-        Learning Research 136:238-253 Available from https://proceedings.mlr.press/v136/mohsenvand20a.html.
-
-    """
-
-    operation = staticmethod(channels_rereference)  # type: ignore[assignment]
-
-    def __init__(self, probability, random_state=None):
-        super().__init__(probability=probability, random_state=random_state)
-
-    def get_augmentation_params(self, *batch):
-        """Return transform parameters"""
-        return {
-            "random_state": self.rng,
-        }
-
-
-class AmplitudeScale(Transform):
-    """Rescale amplitude based on a random sampled scaling value.
-
-    Part of the augmentations proposed in [1]_
-
-    Parameters
-    ----------
-    probability: float
-        Float setting the probability of applying the operation.
-    random_state: int | numpy.random.Generator, optional
-        Seed to be used to instantiate numpy random number generator instance.
-        Used to decide whether or not to transform given the probability
-        argument, to sample which channels to shuffle and to carry the shuffle.
-        Defaults to None.
-
-    References
-    ----------
-    .. [1] Mohsenvand, M.N., Izadi, M.R. &amp; Maes, P.. (2020). Contrastive
-        Representation Learning for Electroencephalogram Classification. Proceedings
-        of the Machine Learning for Health NeurIPS Workshop, in Proceedings of Machine
-        Learning Research 136:238-253 Available from https://proceedings.mlr.press/v136/mohsenvand20a.html.
-
-    """
-
-    operation = staticmethod(amplitude_scale)  # type: ignore[assignment]
-
-    def __init__(self, probability, interval=(0.5, 2), random_state=None):
-        super().__init__(probability=probability, random_state=random_state)
-        self.scale = interval
-
-    def get_augmentation_params(self, *batch):
-        """Return transform parameters"""
-        return {"random_state": self.rng, "scale": self.scale}

{braindecode-1.3.0.dev175887834 → braindecode-1.3.0.dev176089716}/braindecode/classifier.py

@@ -8,6 +8,7 @@
 
 import warnings
 
+import numpy as np
 from skorch import NeuralNet
 from skorch.callbacks import EpochScoring
 from skorch.classifier import NeuralNetClassifier
@@ -236,9 +237,13 @@ class EEGClassifier(_EEGNeuralNet, NeuralNetClassifier):
             num_workers=self.get_iterator(X, training=False).loader.num_workers,
         )
 
-    @property
-    def mode(self):
-        return "classification"
+    def _get_n_outputs(self, y, classes):
+        classes_y = np.unique(y)
+        if classes is not None:
+            assert set(classes_y) <= set(classes)
+        else:
+            classes = classes_y
+        return len(classes)
 
     # Only add the 'accuracy' callback if we are not in cropped mode.
     @property

{braindecode-1.3.0.dev175887834 → braindecode-1.3.0.dev176089716}/braindecode/datasets/__init__.py

@@ -11,11 +11,9 @@ from .base import (
 )
 from .bcicomp import BCICompetitionIVDataset4
 from .bids import BIDSDataset, BIDSEpochsDataset
-from .chb_mit import CHBMIT
 from .mne import create_from_mne_epochs, create_from_mne_raw
 from .moabb import BNCI2014_001, HGD, MOABBDataset
 from .nmt import NMT
-from .siena import SIENA
 from .sleep_physio_challe_18 import SleepPhysionetChallenge2018
 from .sleep_physionet import SleepPhysionet
 from .tuh import TUH, TUHAbnormal
@@ -36,9 +34,7 @@ __all__ = [
     "create_from_mne_epochs",
     "TUH",
     "TUHAbnormal",
-    "SIENA",
     "NMT",
-    "CHBMIT",
     "SleepPhysionet",
     "SleepPhysionetChallenge2018",
     "create_from_X_y",

{braindecode-1.3.0.dev175887834 → braindecode-1.3.0.dev176089716}/braindecode/datasets/base.py

@@ -19,7 +19,7 @@ import warnings
 from abc import abstractmethod
 from collections.abc import Callable
 from glob import glob
-from typing import Any, Generic, Iterable, no_type_check
+from typing import Generic, Iterable, no_type_check
 
 import mne.io
 import numpy as np
@@ -28,9 +28,6 @@ from mne.utils.docs import deprecated
 from torch.utils.data import ConcatDataset, Dataset
 from typing_extensions import TypeVar
 
-from .hub import HubDatasetMixin
-from .registry import register_dataset
-
 
 def _create_description(description) -> pd.Series:
     if description is not None:
@@ -100,7 +97,6 @@ class RecordDataset(Dataset[tuple[np.ndarray, int | str, tuple[int, int, int]]])
 T = TypeVar("T", bound=RecordDataset)
 
 
-@register_dataset
 class RawDataset(RecordDataset):
     """Returns samples from an mne.io.Raw object along with a target.
 
@@ -133,7 +129,6 @@ class RawDataset(RecordDataset):
 
         # save target name for load/save later
         self.target_name = self._target_name(target_name)
-        self.raw_preproc_kwargs: list[dict[str, Any]] = []
 
     def __getitem__(self, index):
         X = self.raw[:, index][0]
@@ -181,12 +176,10 @@ class RawDataset(RecordDataset):
     "If you want to type a Braindecode dataset (i.e. RawDataset|EEGWindowsDataset|WindowsDataset), "
     "use the RecordDataset class instead."
 )
-@register_dataset
 class BaseDataset(RawDataset):
     pass
 
 
-@register_dataset
 class EEGWindowsDataset(RecordDataset):
     """Returns windows from an mne.Raw object, its window indices, along with a target.
 
@@ -242,7 +235,6 @@ class EEGWindowsDataset(RecordDataset):
         ].to_numpy()
         if self.targets_from == "metadata":
             self.y = metadata.loc[:, "target"].to_list()
-        self.raw_preproc_kwargs: list[dict[str, Any]] = []
 
     def __getitem__(self, index: int):
         """Get a window and its target.
@@ -293,7 +285,6 @@ class EEGWindowsDataset(RecordDataset):
         return len(self.crop_inds)
 
 
-@register_dataset
 class WindowsDataset(RecordDataset):
     """Returns windows from an mne.Epochs object along with a target.
 
@@ -343,8 +334,6 @@ class WindowsDataset(RecordDataset):
         ].to_numpy()
         if self.targets_from == "metadata":
             self.y = metadata.loc[:, "target"].to_list()
-        self.raw_preproc_kwargs: list[dict[str, Any]] = []
-        self.window_preproc_kwargs: list[dict[str, Any]] = []
 
     def __getitem__(self, index: int):
         """Get a window and its target.
@@ -385,16 +374,12 @@ class WindowsDataset(RecordDataset):
         return len(self.windows.events)
 
 
-@register_dataset
-class BaseConcatDataset(ConcatDataset, HubDatasetMixin, Generic[T]):
+class BaseConcatDataset(ConcatDataset, Generic[T]):
     """A base class for concatenated datasets.
 
     Holds either mne.Raw or mne.Epoch in self.datasets and has
     a pandas DataFrame with additional description.
 
-    Includes Hugging Face Hub integration via HubDatasetMixin for
-    uploading and downloading datasets.
-
     Parameters
     ----------
     list_of_ds : list
@@ -809,7 +794,7 @@ class BaseConcatDataset(ConcatDataset, HubDatasetMixin, Generic[T]):
                 kwargs = getattr(ds, kwargs_name)
                 if kwargs is not None:
                     with open(kwargs_file_path, "w") as f:
-                        json.dump(kwargs, f, indent=2)
+                        json.dump(kwargs, f)
 
     @staticmethod
     def _save_target_name(sub_dir, ds):

{braindecode-1.3.0.dev175887834 → braindecode-1.3.0.dev176089716}/braindecode/datasets/nmt.py

@@ -32,7 +32,6 @@ from joblib import Parallel, delayed
 from mne.datasets import fetch_dataset
 
 from braindecode.datasets.base import BaseConcatDataset, RawDataset
-from braindecode.datasets.utils import _correct_dataset_path
 
 NMT_URL = "https://zenodo.org/record/10909103/files/NMT.zip"
 NMT_archive_name = "NMT.zip"
@@ -97,34 +96,22 @@ class NMT(BaseConcatDataset):
         preload=False,
         n_jobs=1,
     ):
-        # Convert empty string to None for consistency
-        if path == "":
-            path = None
+        # correct the path if needed
+        if path is not None:
+            list_csv = glob.glob(f"{path}/**/Labels.csv", recursive=True)
+            if isinstance(list_csv, list) and len(list_csv) > 0:
+                path = Path(list_csv[0]).parent
 
-        # Download dataset if not present
-        if path is None:
+        if path is None or len(list_csv) == 0:
             path = fetch_dataset(
                 dataset_params=NMT_dataset_params,
-                path=None,
+                path=Path(path) if path is not None else None,
                 processor="unzip",
                 force_update=False,
             )
             # First time we fetch the dataset, we need to move the files to the
             # correct directory.
-            path = _correct_dataset_path(
-                path, NMT_archive_name, "nmt_scalp_eeg_dataset"
-            )
-        else:
-            # Validate that the provided path is a valid NMT dataset
-            if not Path(f"{path}/Labels.csv").exists():
-                raise ValueError(
-                    f"The provided path {path} does not contain a valid "
-                    "NMT dataset (missing Labels.csv). Please ensure the "
-                    "path points directly to the NMT dataset directory."
-                )
-            path = _correct_dataset_path(
-                path, NMT_archive_name, "nmt_scalp_eeg_dataset"
-            )
+            path = _correct_path(path)
 
         # Get all file paths
         file_paths = glob.glob(
@@ -189,6 +176,39 @@ class NMT(BaseConcatDataset):
         return base_dataset
 
 
+def _correct_path(path: str):
+    """
+    Check if the path is correct and rename the file if needed.
+
+    Parameters
+    ----------
+    path: basestring
+        Path to the file.
+
+    Returns
+    -------
+    path: basestring
+        Corrected path.
+    """
+    if not Path(path).exists():
+        unzip_file_name = f"{NMT_archive_name}.unzip"
+        if (Path(path).parent / unzip_file_name).exists():
+            try:
+                os.rename(
+                    src=Path(path).parent / unzip_file_name,
+                    dst=Path(path),
+                )
+
+            except PermissionError:
+                raise PermissionError(
+                    f"Please rename {Path(path).parent / unzip_file_name}"
+                    + f"manually to {path} and try again."
+                )
+        path = os.path.join(path, "nmt_scalp_eeg_dataset")
+
+    return path
+
+
 def _get_header(*args):
     all_paths = {**_NMT_PATHS}
     return all_paths[args[0]]
@@ -268,33 +288,18 @@ _NMT_PATHS = {
 class _NMTMock(NMT):
     """Mocked class for testing and examples."""
 
-    @mock.patch("pathlib.Path.exists", return_value=True)
-    @mock.patch("braindecode.datasets.nmt._correct_dataset_path")
-    @mock.patch("mne.datasets.fetch_dataset")
-    @mock.patch("pandas.read_csv", new=_fake_pd_read_csv)
-    @mock.patch("mne.io.read_raw_edf", new=_fake_raw)
     @mock.patch("glob.glob", return_value=_NMT_PATHS.keys())
+    @mock.patch("mne.io.read_raw_edf", new=_fake_raw)
+    @mock.patch("pandas.read_csv", new=_fake_pd_read_csv)
     def __init__(
         self,
         mock_glob,
-        mock_fetch,
-        mock_correct_path,
-        mock_path_exists,
         path,
         recording_ids=None,
         target_name="pathological",
         preload=False,
         n_jobs=1,
     ):
-        # Prevent download by providing a dummy path if empty/None
-        if not path:
-            path = "mocked_nmt_path"
-
-        # Mock fetch_dataset to return a valid path without downloading
-        mock_fetch.return_value = path
-        # Mock _correct_dataset_path to return the path as-is
-        mock_correct_path.side_effect = lambda p, *args, **kwargs: p
-
         with warnings.catch_warnings():
             warnings.filterwarnings("ignore", message="Cannot save date file")
             super().__init__(

{braindecode-1.3.0.dev175887834 → braindecode-1.3.0.dev176089716}/braindecode/datautil/__init__.py

@@ -11,7 +11,6 @@ from .serialization import (
     load_concat_dataset,
     save_concat_dataset,
 )
-from .util import infer_signal_properties
 
 
 def __getattr__(name):
@@ -60,5 +59,4 @@ __all__ = [
     "_check_save_dir_empty",
     "match_hemisphere_chans",
     "division_channels_idx",
-    "infer_signal_properties",
 ]

{braindecode-1.3.0.dev175887834 → braindecode-1.3.0.dev176089716}/braindecode/datautil/serialization.py

@@ -138,7 +138,7 @@ def _load_signals(fif_file, preload, is_raw):
         with open(pkl_file, "rb") as f:
             signals = pickle.load(f)
 
-        if all(Path(f).exists() for f in signals.filenames):
+        if all(f.exists() for f in signals.filenames):
             if preload:
                 signals.load_data()
             return signals
@@ -302,6 +302,7 @@ def _load_kwargs_json(kwargs_name, sub_dir):
     kwargs_file_path = os.path.join(sub_dir, kwargs_file_name)
     if os.path.exists(kwargs_file_path):
         kwargs = json.load(open(kwargs_file_path, "r"))
+        kwargs = [tuple(kwarg) for kwarg in kwargs]
         return kwargs