eegdash 0.3.0.dev172943270__py3-none-any.whl → 0.3.1__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.0.dev172943270"
+__version__ = "0.3.1"

eegdash/data_utils.py

@@ -61,10 +61,23 @@ class EEGDashBaseDataset(BaseDataset):
             suffix="eeg",
             **bids_kwargs,
         )
-        self.s3_bucket = s3_bucket if s3_bucket else self._AWS_BUCKET
+        if s3_bucket:
+            self.s3_bucket = s3_bucket
+            self.s3_open_neuro = False
+        else:
+            self.s3_bucket = self._AWS_BUCKET
+            self.s3_open_neuro = True
+
         self.s3file = self.get_s3path(record["bidspath"])
         self.filecache = self.cache_dir / record["bidspath"]
         self.bids_dependencies = record["bidsdependencies"]
+        # Temporary fix for BIDS dependencies path
+        # just to release to the competition
+        if not self.s3_open_neuro:
+            self.bids_dependencies = [
+                dep.split("/", 1)[1] for dep in self.bids_dependencies
+            ]
+
         self._raw = None
 
     def get_s3path(self, filepath: str) -> str:
@@ -77,6 +90,9 @@ class EEGDashBaseDataset(BaseDataset):
         filesystem = s3fs.S3FileSystem(
             anon=True, client_kwargs={"region_name": "us-east-2"}
         )
+        if not self.s3_open_neuro:
+            self.s3file = re.sub(r"(^|/)ds\d{6}/", r"\1", self.s3file, count=1)
+
         filesystem.download(self.s3file, self.filecache)
         self.filenames = [self.filecache]
 
@@ -108,7 +124,7 @@ class EEGDashBaseDataset(BaseDataset):
                 self._download_dependencies()
             self._download_s3()
         if self._raw is None:
-            self._raw = mne.io.read_raw(self.bidspath, verbose=False)
+            self._raw = mne.io.read_raw(fname=self.bidspath, verbose=False)
 
     # === BaseDataset and PyTorch Dataset interface ===
 
@@ -215,6 +231,7 @@ class EEGDashBaseRaw(BaseRaw):
         )
 
     def get_s3path(self, filepath):
+        print(f"Getting S3 path for {filepath}")
         return f"{self._AWS_BUCKET}/{filepath}"
 
     def _download_s3(self):

eegdash/dataset.py

@@ -1,5 +1,19 @@
 from .api import EEGDashDataset
 
+RELEASE_TO_OPENNEURO_DATASET_MAP = {
+    "R11": "ds005516",
+    "R10": "ds005515",
+    "R9": "ds005514",
+    "R8": "ds005512",
+    "R7": "ds005511",
+    "R6": "ds005510",
+    "R4": "ds005508",
+    "R5": "ds005509",
+    "R3": "ds005507",
+    "R2": "ds005506",
+    "R1": "ds005505",
+}
+
 
 class EEGChallengeDataset(EEGDashDataset):
     def __init__(
@@ -7,7 +21,7 @@ class EEGChallengeDataset(EEGDashDataset):
         release: str = "R5",
         query: dict | None = None,
         cache_dir: str = ".eegdash_cache",
-        s3_bucket: str | None = "s3://nmdatasets/NeurIPS25/",
+        s3_bucket: str | None = "s3://nmdatasets/NeurIPS25",
         **kwargs,
     ):
         """Create a new EEGDashDataset from a given query or local BIDS dataset directory
@@ -32,25 +46,11 @@ class EEGChallengeDataset(EEGDashDataset):
             constructor.
 
         """
-        dsnumber_release_map = {
-            "R11": "ds005516",
-            "R10": "ds005515",
-            "R9": "ds005514",
-            "R8": "ds005512",
-            "R7": "ds005511",
-            "R6": "ds005510",
-            "R4": "ds005508",
-            "R5": "ds005509",
-            "R3": "ds005507",
-            "R2": "ds005506",
-            "R1": "ds005505",
-        }
-
         self.release = release
-        if release not in dsnumber_release_map:
+        if release not in RELEASE_TO_OPENNEURO_DATASET_MAP:
             raise ValueError(f"Unknown release: {release}")
 
-        dataset = dsnumber_release_map[release]
+        dataset = RELEASE_TO_OPENNEURO_DATASET_MAP[release]
         if query is None:
             query = {"dataset": dataset}
         elif "dataset" not in query:

{eegdash-0.3.0.dev172943270.dist-info → eegdash-0.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: eegdash
-Version: 0.3.0.dev172943270
+Version: 0.3.1
 Summary: EEG data for machine learning
 Author-email: Young Truong <dt.young112@gmail.com>, Arnaud Delorme <adelorme@gmail.com>, Bruno Aristimunha <b.aristimunha@gmail.com>
 License: GNU General Public License

{eegdash-0.3.0.dev172943270.dist-info → eegdash-0.3.1.dist-info}/RECORD

@@ -1,8 +1,8 @@
-eegdash/__init__.py,sha256=_DxCP6T8R_-lnyPltTh3ERkX5Dlp4dqMJ9VMfeEylS8,245
+eegdash/__init__.py,sha256=sNP5x1-Ek6HhCM9KzACR7_WvkY7NYvvw3Iv6um43gfw,232
 eegdash/api.py,sha256=igpkLTCjH6NNQf8tEizA6SF8cYa1gE22-hNVxNRPszg,26747
 eegdash/data_config.py,sha256=OS6ERO-jHrnEOfMJUehY7ieABdsRw_qWzOKJ4pzSfqw,1323
-eegdash/data_utils.py,sha256=JN7JFaFg1qY0Cnmhnan5RKMn7j3OOatdiio6jZ8NAhQ,23635
-eegdash/dataset.py,sha256=NK5CVGtHvGNlhdbZPaRUWtyLBDxUlyiSz-ot3BrxHQE,2443
+eegdash/data_utils.py,sha256=hUjSO-2hBZF_f7We7RSXbm3H15-j58rBZZxmFFggyJE,24191
+eegdash/dataset.py,sha256=BngXb9TtAo0Sjc3o5bRC0fJkQrThdsqzx2Zpvp7xtvs,2374
 eegdash/mongodb.py,sha256=GD3WgA253oFgpzOHrYaj4P1mRjNtDMT5Oj4kVvHswjI,2006
 eegdash/preprocessing.py,sha256=7S_TTRKPKEk47tTnh2D6WExBt4cctAMxUxGDjJqq5lU,2221
 eegdash/utils.py,sha256=wU9CBQZLW_LIQIBwhgQm5bU4X-rSsVNPdeF2iE4QGJ4,410
@@ -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.0.dev172943270.dist-info/licenses/LICENSE,sha256=KykUD4H3kw3HLz5bZ0kxMWwZotnk8rhkfCCerGyX2sk,855
-eegdash-0.3.0.dev172943270.dist-info/METADATA,sha256=ubGyqDkjsgvRcDX0hGrZfXXZk2J5am8d9yA0Whc4OsU,10634
-eegdash-0.3.0.dev172943270.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-eegdash-0.3.0.dev172943270.dist-info/top_level.txt,sha256=zavO69HQ6MyZM0aQMR2zUS6TAFc7bnN5GEpDpOpFZzU,8
-eegdash-0.3.0.dev172943270.dist-info/RECORD,,
+eegdash-0.3.1.dist-info/licenses/LICENSE,sha256=KykUD4H3kw3HLz5bZ0kxMWwZotnk8rhkfCCerGyX2sk,855
+eegdash-0.3.1.dist-info/METADATA,sha256=XPr6_yorRXXsHSRWILffJ09I9yrLfs8m_36Gd7-mIZY,10621
+eegdash-0.3.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+eegdash-0.3.1.dist-info/top_level.txt,sha256=zavO69HQ6MyZM0aQMR2zUS6TAFc7bnN5GEpDpOpFZzU,8
+eegdash-0.3.1.dist-info/RECORD,,