Braindecode 1.5.0.dev1004__py3-none-any.whl → 1.5.0.dev1007__py3-none-any.whl

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):

braindecode/preprocessing/preprocess.py

@@ -6,7 +6,6 @@
 #          David Sabbagh <dav.sabbagh@gmail.com>
 #          Bruno Aristimunha <b.aristimunha@gmail.com>
 #          Léo Burgund <leo.burgund@gmail.com>
-#          Sarthak Tayal <sarthaktayal2@gmail.com>
 #
 # License: BSD (3-clause)
 
@@ -218,7 +217,6 @@ def preprocess(
     offset: int = 0,
     copy_data: bool | None = None,
     parallel_kwargs: dict | None = None,
-    max_nbytes: int | str | None = "1M",
 ):
     """Apply preprocessors to a concat dataset.
 
@@ -246,20 +244,14 @@ def preprocess(
         Additional keyword arguments forwarded to ``joblib.Parallel``.
         Defaults to None (equivalent to ``{}``).
         See https://joblib.readthedocs.io/en/stable/generated/joblib.Parallel.html for details.
-    max_nbytes : int, str, or None
-        Threshold (in bytes; or e.g. ``"1M"``) above which joblib memory-maps
-        preloaded arrays as read-only when dispatching to worker processes.
-        Effective only when ``n_jobs != 1``. Pass ``None`` to disable memory
-        mapping when a preprocessor resizes the underlying data (for example
-        ``filterbank``), which would otherwise fail with an ``mmap can't
-        resize a readonly`` error. ``parallel_kwargs['max_nbytes']`` takes
-        precedence if both are provided.
 
     Returns
     -------
     BaseConcatDataset
         Preprocessed dataset.
     """
+    # In case of serialization, make sure directory is available before
+    # preprocessing
     if save_dir is not None and not overwrite:
         _check_save_dir_empty(save_dir)
 
@@ -274,35 +266,22 @@ def preprocess(
     parallel_params.setdefault(
         "prefer", "threads" if platform.system() == "Windows" else None
     )
-    parallel_params.setdefault("max_nbytes", max_nbytes)
-
-    try:
-        list_of_ds = Parallel(n_jobs=n_jobs, **parallel_params)(
-            delayed(_preprocess)(
-                ds,
-                i + offset,
-                preprocessors,
-                save_dir,
-                overwrite,
-                copy_data=(
-                    (parallel_processing and (save_dir is None))
-                    if copy_data is None
-                    else copy_data
-                ),
-            )
-            for i, ds in enumerate(concat_ds.datasets)
+
+    list_of_ds = Parallel(n_jobs=n_jobs, **parallel_params)(
+        delayed(_preprocess)(
+            ds,
+            i + offset,
+            preprocessors,
+            save_dir,
+            overwrite,
+            copy_data=(
+                (parallel_processing and (save_dir is None))
+                if copy_data is None
+                else copy_data
+            ),
         )
-    except (BufferError, ValueError, OSError) as exc:
-        msg = str(exc).lower().replace("-", "")
-        if "mmap" in msg and "readonly" in msg:
-            raise RuntimeError(
-                "Parallel preprocessing failed because joblib memory-mapped "
-                "a preloaded array that a preprocessor then attempted to "
-                "resize (e.g. ``filterbank``). Pass ``max_nbytes=None`` to "
-                "``preprocess`` to disable memory mapping, or supply a "
-                "``save_dir`` so the data is reloaded with ``preload=False``."
-            ) from exc
-        raise
+        for i, ds in enumerate(concat_ds.datasets)
+    )
 
     if save_dir is not None:  # Reload datasets and replace in concat_ds
         ids_to_load = [i + offset for i in range(len(concat_ds.datasets))]

braindecode/version.py

@@ -1 +1 @@
-__version__ = "1.5.0.dev1004"
+__version__ = "1.5.0.dev1007"

{braindecode-1.5.0.dev1004.dist-info → braindecode-1.5.0.dev1007.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: braindecode
-Version: 1.5.0.dev1004
+Version: 1.5.0.dev1007
 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.0.dev1004.dist-info → braindecode-1.5.0.dev1007.dist-info}/RECORD

@@ -3,7 +3,7 @@ braindecode/classifier.py,sha256=7kC_oY_UzHEes_WWdCvEpiA1ZKxMeuLL5tIPp5rfcpg,962
 braindecode/eegneuralnet.py,sha256=xjE6aPZdCQPs29NIpy_m1GLMMC2WZ3Db0Fuh1-xE1h4,13827
 braindecode/regressor.py,sha256=KiMJpqCUPWA2k2JWk9HGYTzeoBqJ4gAKEudeUVcFZY4,9266
 braindecode/util.py,sha256=f8bNIwt-SwsHqheH_BADQxTtA9oPt3Lb7GFnoI-Huwc,14101
-braindecode/version.py,sha256=Jctehl91ExJ5Idkg5_Gh4V4SK7S_G6mt0SO_eABfZFs,30
+braindecode/version.py,sha256=D3ZA75b03PZQ80vQOvTo6EESMVaElofg2dXgccZLJV8,30
 braindecode/augmentation/__init__.py,sha256=4xune2QUK6KHMKsAqijF7I9eeiVbP0wEoQJjCNLNcKM,1081
 braindecode/augmentation/base.py,sha256=OJ1shOljI1yTY9zh2qWxQwivlY43sfx9Q-MAyMhxtPs,7338
 braindecode/augmentation/functional.py,sha256=q2k6mAXrujYlOZUndcjZN8e8b-6oJF1gGsORAI23hyE,43998
@@ -43,7 +43,7 @@ braindecode/models/__init__.py,sha256=SwnwJ-nRF0nWq2YYvw6G4F_zMrw0QCQW2VoYE2KOpS
 braindecode/models/atcnet.py,sha256=JzLV92WYWlhQol8AqE9xeWvlnWN_uswYhVOkMnuBCXo,32231
 braindecode/models/attentionbasenet.py,sha256=k7ar7aEjANudPu7krAZsRx-ag61ugirS1Xn7qFKNfWg,30483
 braindecode/models/attn_sleep.py,sha256=F9x4spTtzfiCC1h9UYITmIDQeJW6_2CXTZktZX9R0RE,17950
-braindecode/models/base.py,sha256=O3pkmdqaBgLWHpWkVF7ld0A95T43RFyno7mtkyi08as,28327
+braindecode/models/base.py,sha256=yGJgr0f5rD-gJZ5Msw9FzGzVO-x_re-6jOSK8Iht6x4,27923
 braindecode/models/bendr.py,sha256=RkPeHoFF0vcDBW2jMo_9oS1Pxuq3wcI4ErDeL7L8QGs,29157
 braindecode/models/biot.py,sha256=UPpT1Gv7siBCMcNDNx4DS4hgs-YlmWwU5Nmikaj9WpY,22309
 braindecode/models/brainmodule.py,sha256=idyQVTp3VBJXKF1YjMx8o1kUKrcL_E_AJsioPjcuqV8,33282
@@ -111,7 +111,7 @@ braindecode/modules/wrapper.py,sha256=3lgNjcwJ1Kre2TCUouKpEp6cidFgp6LBDDWQNLOvRQ
 braindecode/preprocessing/__init__.py,sha256=NCKhT4-_-DMBFpppEg2VguCe4x7ZW-gInWn60G6hOSs,9643
 braindecode/preprocessing/eegprep_preprocess.py,sha256=wGKKo-JvG90n7eB7Y_p-bSkf8LAyIxTU8N01qPjAYQI,59989
 braindecode/preprocessing/mne_preprocess.py,sha256=taW2H-k3yreGdItCW7ldZB8jnzxBAMxX1ykgpvj7QvI,8246
-braindecode/preprocessing/preprocess.py,sha256=6qm0tU5JZjCLEl125cflaDk-B54rMsN8JbUUQG6cWW0,23774
+braindecode/preprocessing/preprocess.py,sha256=T-MA0frxrCb61U30F9_LoljyXD5oUHgnI3uWiggfgH8,22515
 braindecode/preprocessing/util.py,sha256=ivODABSuy-SKvPMa2U6W3uWM4cwmSg-7jSKqIRxBBw4,5925
 braindecode/preprocessing/windowers.py,sha256=NTplG_8pOUrE1R-w41vevuBK_4_RHEozy1iR3RyoNCU,48949
 braindecode/samplers/__init__.py,sha256=TLuO6gXv2WioJdX671MI_CHVSsOfbjnly1Xv9K3_WdA,452
@@ -128,9 +128,9 @@ braindecode/visualization/frequency.py,sha256=gNwkn9yIik5SUp7d9HE9J_vPVGyzNsxxCO
 braindecode/visualization/metrics.py,sha256=j01kc04P9uEkQ2g2Tt2C76yr6soIj31PAuBMflrmODg,13615
 braindecode/visualization/sanity.py,sha256=nNClauUC8dCj_KCy_1RmaPDQAqExLczfPtUeQ7k9-Q0,4812
 braindecode/visualization/topology.py,sha256=mXxUfCCUJqa_cMF4y6GC3_A-qBCcS4uTc0EzBolkytE,2274
-braindecode-1.5.0.dev1004.dist-info/licenses/LICENSE.txt,sha256=7rg7k6hyj8m9whQ7dpKbqnCssoOEx_Mbtqb4uSOjljE,1525
-braindecode-1.5.0.dev1004.dist-info/licenses/NOTICE.txt,sha256=ZFFhigxIaKgDcMjCzPyAVSFV42ztU0kLOENt_kvherw,857
-braindecode-1.5.0.dev1004.dist-info/METADATA,sha256=wET5lgMnARymy9-u-josg-o6vbHYlOBTknou-H8bx9U,10275
-braindecode-1.5.0.dev1004.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
-braindecode-1.5.0.dev1004.dist-info/top_level.txt,sha256=pHsWQmSy0uhIez62-HA9j0iaXKvSbUL39ifFRkFnChA,12
-braindecode-1.5.0.dev1004.dist-info/RECORD,,
+braindecode-1.5.0.dev1007.dist-info/licenses/LICENSE.txt,sha256=7rg7k6hyj8m9whQ7dpKbqnCssoOEx_Mbtqb4uSOjljE,1525
+braindecode-1.5.0.dev1007.dist-info/licenses/NOTICE.txt,sha256=ZFFhigxIaKgDcMjCzPyAVSFV42ztU0kLOENt_kvherw,857
+braindecode-1.5.0.dev1007.dist-info/METADATA,sha256=fxnOI00CSFVaUb9jrwzxuNGBpOsqlvIiwIMNg1-gf2I,10275
+braindecode-1.5.0.dev1007.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+braindecode-1.5.0.dev1007.dist-info/top_level.txt,sha256=pHsWQmSy0uhIez62-HA9j0iaXKvSbUL39ifFRkFnChA,12
+braindecode-1.5.0.dev1007.dist-info/RECORD,,