eegdash 0.3.3.dev61__tar.gz → 0.3.3.dev67__tar.gz

{eegdash-0.3.3.dev61 → eegdash-0.3.3.dev67}/PKG-INFO

@@ -1,35 +1,11 @@
 Metadata-Version: 2.4
 Name: eegdash
-Version: 0.3.3.dev61
+Version: 0.3.3.dev67
 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
-        
-        Copyright (C) 2024-2025
-        
-        Young Truong, UCSD, dt.young112@gmail.com
-        Arnaud Delorme, UCSD, adelorme@ucsd.edu
-        Aviv Dotan, BGU, avivdot@bgu.post.ac.il
-        Oren Shriki, BGU, shrikio@bgu.ac.il
-        Bruno Aristimunha, b.aristimunha@gmail.com
-        
-        This program is free software; you can redistribute it and/or modify
-        it under the terms of the GNU General Public License as published by
-        the Free Software Foundation; either version 2 of the License, or
-        (at your option) any later version.
-        
-        This program is distributed in the hope that it will be useful,
-        but WITHOUT ANY WARRANTY; without even the implied warranty of
-        MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-        GNU General Public License for more details.
-        
-        You should have received a copy of the GNU General Public License
-        along with this program; if not, write to the Free Software
-        Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1.07  USA
-        
+License-Expression: GPL-3.0-only
 Project-URL: Homepage, https://github.com/sccn/EEG-Dash-Data
 Project-URL: Issues, https://github.com/sccn/EEG-Dash-Data/issues
-Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Classifier: Intended Audience :: Science/Research
 Classifier: Intended Audience :: Developers

{eegdash-0.3.3.dev61 → eegdash-0.3.3.dev67}/eegdash/__init__.py

@@ -4,5 +4,7 @@ from .utils import __init__mongo_client
 
 __init__mongo_client()
 
+
 __all__ = ["EEGDash", "EEGDashDataset", "EEGChallengeDataset"]
-__version__ = "0.3.3.dev61"
+
+__version__ = "0.3.3.dev67"

{eegdash-0.3.3.dev61 → eegdash-0.3.3.dev67}/eegdash/data_utils.py

@@ -2,10 +2,12 @@ import json
 import logging
 import os
 import re
+import warnings
 from pathlib import Path
 from typing import Any
 
 import mne
+import mne_bids
 import numpy as np
 import pandas as pd
 import s3fs
@@ -174,7 +176,16 @@ class EEGDashBaseDataset(BaseDataset):
                 self._download_dependencies()
             self._download_s3()
         if self._raw is None:
-            self._raw = mne.io.read_raw(fname=self.bidspath, verbose=False)
+            # capturing any warnings
+            # to-do: remove this once is fixed on the mne-bids side.
+            with warnings.catch_warnings(record=True) as w:
+                self._raw = mne_bids.read_raw_bids(
+                    bids_path=self.bidspath, verbose="ERROR"
+                )
+                for warning in w:
+                    logger.warning(
+                        f"Warning while reading BIDS file: {warning.message}"
+                    )
 
     # === BaseDataset and PyTorch Dataset interface ===
 

{eegdash-0.3.3.dev61 → eegdash-0.3.3.dev67}/eegdash/dataset.py

@@ -1,4 +1,7 @@
+from pathlib import Path
+
 from .api import EEGDashDataset
+from .registry import register_openneuro_datasets
 
 RELEASE_TO_OPENNEURO_DATASET_MAP = {
     "R11": "ds005516",
@@ -67,3 +70,13 @@ class EEGChallengeDataset(EEGDashDataset):
             s3_bucket=f"{s3_bucket}/{release}_L100",
             **kwargs,
         )
+
+
+registered_classes = register_openneuro_datasets(
+    summary_file=Path(__file__).with_name("dataset_summary.csv"),
+    base_class=EEGDashDataset,
+    namespace=globals(),
+)
+
+
+__all__ = ["EEGChallengeDataset"] + list(registered_classes.keys())

eegdash-0.3.3.dev67/eegdash/registry.py

@@ -0,0 +1,72 @@
+from __future__ import annotations
+
+import csv
+from pathlib import Path
+from typing import Any, Dict
+
+
+def register_openneuro_datasets(
+    summary_file: str | Path,
+    *,
+    base_class=None,
+    namespace: Dict[str, Any] | None = None,
+) -> Dict[str, type]:
+    """Dynamically create dataset classes from a summary file.
+
+    Parameters
+    ----------
+    summary_file : str | Path
+        Path to a CSV file where each line starts with the dataset identifier.
+    base_class : type | None
+        Base class for the generated datasets. If ``None``, defaults to
+        :class:`eegdash.api.EEGDashDataset`.
+    namespace : dict | None
+        Mapping where the new classes will be registered. Defaults to the
+        module's global namespace.
+
+    Returns
+    -------
+    dict
+        Mapping from class names to the generated classes.
+
+    """
+    if base_class is None:
+        from .api import EEGDashDataset as base_class  # lazy import
+
+    summary_path = Path(summary_file)
+    namespace = namespace if namespace is not None else globals()
+    registered: Dict[str, type] = {}
+
+    with summary_path.open() as f:
+        reader = csv.reader(f)
+        for row in reader:
+            if not row:
+                continue
+            dataset_id = row[0].strip()
+            if not dataset_id or dataset_id.startswith("#"):
+                continue
+            class_name = dataset_id.upper()
+
+            def __init__(
+                self,
+                cache_dir: str,
+                query: dict | None = None,
+                s3_bucket: str | None = None,
+                **kwargs,
+            ):
+                q = {"dataset": self._dataset}
+                if query:
+                    q.update(query)
+                super().__init__(
+                    query=q, cache_dir=cache_dir, s3_bucket=s3_bucket, **kwargs
+                )
+
+            cls = type(
+                class_name,
+                (base_class,),
+                {"_dataset": dataset_id, "__init__": __init__},
+            )
+            namespace[class_name] = cls
+            registered[class_name] = cls
+
+    return registered

{eegdash-0.3.3.dev61 → eegdash-0.3.3.dev67}/eegdash.egg-info/PKG-INFO

@@ -1,35 +1,11 @@
 Metadata-Version: 2.4
 Name: eegdash
-Version: 0.3.3.dev61
+Version: 0.3.3.dev67
 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
-        
-        Copyright (C) 2024-2025
-        
-        Young Truong, UCSD, dt.young112@gmail.com
-        Arnaud Delorme, UCSD, adelorme@ucsd.edu
-        Aviv Dotan, BGU, avivdot@bgu.post.ac.il
-        Oren Shriki, BGU, shrikio@bgu.ac.il
-        Bruno Aristimunha, b.aristimunha@gmail.com
-        
-        This program is free software; you can redistribute it and/or modify
-        it under the terms of the GNU General Public License as published by
-        the Free Software Foundation; either version 2 of the License, or
-        (at your option) any later version.
-        
-        This program is distributed in the hope that it will be useful,
-        but WITHOUT ANY WARRANTY; without even the implied warranty of
-        MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
-        GNU General Public License for more details.
-        
-        You should have received a copy of the GNU General Public License
-        along with this program; if not, write to the Free Software
-        Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1.07  USA
-        
+License-Expression: GPL-3.0-only
 Project-URL: Homepage, https://github.com/sccn/EEG-Dash-Data
 Project-URL: Issues, https://github.com/sccn/EEG-Dash-Data/issues
-Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Classifier: Intended Audience :: Science/Research
 Classifier: Intended Audience :: Developers

{eegdash-0.3.3.dev61 → eegdash-0.3.3.dev67}/eegdash.egg-info/SOURCES.txt

@@ -8,6 +8,7 @@ eegdash/data_utils.py
 eegdash/dataset.py
 eegdash/mongodb.py
 eegdash/preprocessing.py
+eegdash/registry.py
 eegdash/utils.py
 eegdash.egg-info/PKG-INFO
 eegdash.egg-info/SOURCES.txt
@@ -31,6 +32,7 @@ eegdash/features/feature_bank/spectral.py
 eegdash/features/feature_bank/utils.py
 tests/test_correctness.py
 tests/test_dataset.py
+tests/test_dataset_registration.py
 tests/test_eegdash.py
 tests/test_functional.py
 tests/test_init.py

{eegdash-0.3.3.dev61 → eegdash-0.3.3.dev67}/pyproject.toml

@@ -14,11 +14,10 @@ authors = [
 ]
 description     = "EEG data for machine learning"
 readme          = "README.md"
-license         = { file = "LICENSE" }
+license         = "GPL-3.0-only"
 requires-python = ">=3.10"
 
 classifiers = [
-    "License :: OSI Approved :: MIT License",
     "Operating System :: OS Independent",
     "Intended Audience :: Science/Research",
     "Intended Audience :: Developers",

eegdash-0.3.3.dev67/tests/test_dataset_registration.py

@@ -0,0 +1,33 @@
+import importlib.util
+from pathlib import Path
+
+
+class DummyBase:
+    pass
+
+
+def test_register_openneuro_datasets(tmp_path: Path):
+    module_path = Path(__file__).resolve().parents[1] / "eegdash" / "registry.py"
+    spec = importlib.util.spec_from_file_location("registry", module_path)
+    registry = importlib.util.module_from_spec(spec)
+    spec.loader.exec_module(registry)
+
+    summary = tmp_path / "dataset_summary.csv"
+    summary.write_text(
+        "\n".join(
+            [
+                "ds002718,18,18,1,74,250,14.844",
+                "ds000001,1,1,1,1,1,1",
+            ]
+        )
+    )
+    namespace = {}
+    registered = registry.register_openneuro_datasets(
+        summary, namespace=namespace, base_class=DummyBase
+    )
+
+    assert set(registered) == {"DS002718", "DS000001"}
+    ds_class = registered["DS002718"]
+    assert ds_class is namespace["DS002718"]
+    assert issubclass(ds_class, DummyBase)
+    assert ds_class._dataset == "ds002718"