eegdash 0.3.2.dev52__py3-none-any.whl → 0.3.3.dev58__py3-none-any.whl

eegdash/__init__.py

@@ -5,4 +5,4 @@ from .utils import __init__mongo_client
 __init__mongo_client()
 
 __all__ = ["EEGDash", "EEGDashDataset", "EEGChallengeDataset"]
-__version__ = "0.3.2.dev52"
+__version__ = "0.3.3.dev58"

eegdash/data_utils.py

@@ -10,6 +10,7 @@ import numpy as np
 import pandas as pd
 import s3fs
 from bids import BIDSLayout
+from fsspec.callbacks import TqdmCallback
 from joblib import Parallel, delayed
 from mne._fiff.utils import _read_segments_file
 from mne.io import BaseRaw
@@ -98,8 +99,27 @@ class EEGDashBaseDataset(BaseDataset):
             self.s3file = re.sub(r"(^|/)ds\d{6}/", r"\1", self.s3file, count=1)
 
         self.filecache.parent.mkdir(parents=True, exist_ok=True)
+        info = filesystem.info(self.s3file)
+        size = info.get("size") or info.get("Size")
+
+        callback = TqdmCallback(
+            size=size,
+            tqdm_kwargs=dict(
+                desc=f"Downloading {Path(self.s3file).name}",
+                unit="B",
+                unit_scale=True,
+                unit_divisor=1024,
+                dynamic_ncols=True,
+                leave=True,
+                mininterval=0.2,
+                smoothing=0.1,
+                miniters=1,
+                bar_format="{desc}: {percentage:3.0f}%|{bar}| {n_fmt}/{total_fmt} "
+                "[{elapsed}<{remaining}, {rate_fmt}]",
+            ),
+        )
+        filesystem.get(self.s3file, self.filecache, callback=callback)
 
-        filesystem.download(self.s3file, self.filecache)
         self.filenames = [self.filecache]
 
     def _download_dependencies(self) -> None:
@@ -119,7 +139,26 @@ class EEGDashBaseDataset(BaseDataset):
             # in the case of the competition.
             if not filepath.exists():
                 filepath.parent.mkdir(parents=True, exist_ok=True)
-                filesystem.download(s3path, filepath)
+                info = filesystem.info(s3path)
+                size = info.get("size") or info.get("Size")
+
+                callback = TqdmCallback(
+                    size=size,
+                    tqdm_kwargs=dict(
+                        desc=f"Downloading {Path(s3path).name}",
+                        unit="B",
+                        unit_scale=True,
+                        unit_divisor=1024,
+                        dynamic_ncols=True,
+                        leave=True,
+                        mininterval=0.2,
+                        smoothing=0.1,
+                        miniters=1,
+                        bar_format="{desc}: {percentage:3.0f}%|{bar}| {n_fmt}/{total_fmt} "
+                        "[{elapsed}<{remaining}, {rate_fmt}]",
+                    ),
+                )
+                filesystem.get(s3path, filepath, callback=callback)
 
     def get_raw_bids_args(self) -> dict[str, Any]:
         """Helper to restrict the metadata record to the fields needed to locate a BIDS

{eegdash-0.3.2.dev52.dist-info → eegdash-0.3.3.dev58.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: eegdash
-Version: 0.3.2.dev52
+Version: 0.3.3.dev58
 Summary: EEG data for machine learning
 Author-email: Young Truong <dt.young112@gmail.com>, Arnaud Delorme <adelorme@gmail.com>, Aviv Dotan <avivd220@gmail.com>, Oren Shriki <oren70@gmail.com>, Bruno Aristimunha <b.aristimunha@gmail.com>
 License: GNU General Public License
@@ -60,6 +60,8 @@ Requires-Dist: s3fs
 Requires-Dist: scipy
 Requires-Dist: tqdm
 Requires-Dist: xarray
+Requires-Dist: h5io>=0.2.4
+Requires-Dist: pymatreader
 Provides-Extra: tests
 Requires-Dist: pytest; extra == "tests"
 Requires-Dist: pytest-cov; extra == "tests"
@@ -98,6 +100,14 @@ Dynamic: license-file
 
 # EEG-Dash
 
+[![PyPI version](https://img.shields.io/pypi/v/eegdash)](https://pypi.org/project/eegdash/)
+[![Docs](https://img.shields.io/badge/docs-stable-brightgreen.svg)](https://sccn.github.io/eegdash)
+
+[![License: GPL-2.0-or-later](https://img.shields.io/badge/License-GPL--2.0--or--later-blue.svg)](LICENSE)
+[![Python versions](https://img.shields.io/pypi/pyversions/eegdash.svg)](https://pypi.org/project/eegdash/)
+[![Downloads](https://pepy.tech/badge/eegdash)](https://pepy.tech/project/eegdash)
+<!-- [![Coverage](https://img.shields.io/codecov/c/github/sccn/eegdash)](https://codecov.io/gh/sccn/eegdash) -->
+
 To leverage recent and ongoing advancements in large-scale computational methods and to ensure the preservation of scientific data generated from publicly funded research, the EEG-DaSh data archive will create a data-sharing resource for MEEG (EEG, MEG) data contributed by collaborators for machine learning (ML) and deep learning (DL) applications. 
 
 ## Data source
@@ -143,7 +153,9 @@ To use the data from a single subject, enter:
 from eegdash import EEGDashDataset
 
 ds_NDARDB033FW5 = EEGDashDataset(
-    {"dataset": "ds005514", "task": "RestingState", "subject": "NDARDB033FW5"}
+    {"dataset": "ds005514", "task":
+     "RestingState", "subject": "NDARDB033FW5"}, 
+     cache_dir="."
 )
 ```
 
@@ -155,7 +167,7 @@ To use the data from multiple subjects, enter:
 from eegdash import EEGDashDataset
 
 ds_ds005505rest = EEGDashDataset(
-    {"dataset": "ds005505", "task": "RestingState"}, target_name="sex"
+    {"dataset": "ds005505", "task": "RestingState"}, target_name="sex", cache_dir=".
 )
 ```
 

{eegdash-0.3.2.dev52.dist-info → eegdash-0.3.3.dev58.dist-info}/RECORD

@@ -1,7 +1,7 @@
-eegdash/__init__.py,sha256=Nfsbk31HxH9sVIE7p14a0SwBKJWCAbWc6pHCUzDy_VA,238
+eegdash/__init__.py,sha256=Kj9Utn1GX3_PBPX9iOkMXepsLn4-u9yx8dFGTIkTb2c,238
 eegdash/api.py,sha256=lYCILa69Y_RRd4_13x1Ge77FDGswnEG2DEfdPzMygAY,26865
 eegdash/data_config.py,sha256=OS6ERO-jHrnEOfMJUehY7ieABdsRw_qWzOKJ4pzSfqw,1323
-eegdash/data_utils.py,sha256=h1FLCTjRj2JDn6IFnVgWrogDYnHFR4mcuDF23KNtLZI,24530
+eegdash/data_utils.py,sha256=8Jb_94uVbdknNPpx3GBl4dCDYUIJNzl3zkLwbfH90N4,26052
 eegdash/dataset.py,sha256=GVELU-eXq9AQDzOeg6Lkykd-Pctyn42e5UEcAV0Go4s,2348
 eegdash/mongodb.py,sha256=GD3WgA253oFgpzOHrYaj4P1mRjNtDMT5Oj4kVvHswjI,2006
 eegdash/preprocessing.py,sha256=7S_TTRKPKEk47tTnh2D6WExBt4cctAMxUxGDjJqq5lU,2221
@@ -21,8 +21,8 @@ eegdash/features/feature_bank/dimensionality.py,sha256=j_Ds71Y1AbV2uLFQj8EuXQ4kz
 eegdash/features/feature_bank/signal.py,sha256=3Tb8z9gX7iZipxQJ9DSyy30JfdmW58kgvimSyZX74p8,3404
 eegdash/features/feature_bank/spectral.py,sha256=bNB7skusePs1gX7NOU6yRlw_Gr4UOCkO_ylkCgybzug,3319
 eegdash/features/feature_bank/utils.py,sha256=DGh-Q7-XFIittP7iBBxvsJaZrlVvuY5mw-G7q6C-PCI,1237
-eegdash-0.3.2.dev52.dist-info/licenses/LICENSE,sha256=asisR-xupy_NrQBFXnx6yqXeZcYWLvbAaiETl25iXT0,931
-eegdash-0.3.2.dev52.dist-info/METADATA,sha256=vlhnJbk-mBMlfsTen_2zpBRpm6VNZ_2gI3I2gprYAC0,10784
-eegdash-0.3.2.dev52.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-eegdash-0.3.2.dev52.dist-info/top_level.txt,sha256=zavO69HQ6MyZM0aQMR2zUS6TAFc7bnN5GEpDpOpFZzU,8
-eegdash-0.3.2.dev52.dist-info/RECORD,,
+eegdash-0.3.3.dev58.dist-info/licenses/LICENSE,sha256=asisR-xupy_NrQBFXnx6yqXeZcYWLvbAaiETl25iXT0,931
+eegdash-0.3.3.dev58.dist-info/METADATA,sha256=RRZw0l9aGFsfVZgS6qfQVTft6uBdVdRqTgsQ6HYjl9A,11483
+eegdash-0.3.3.dev58.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+eegdash-0.3.3.dev58.dist-info/top_level.txt,sha256=zavO69HQ6MyZM0aQMR2zUS6TAFc7bnN5GEpDpOpFZzU,8
+eegdash-0.3.3.dev58.dist-info/RECORD,,