eegdash 0.3.3.dev61__tar.gz → 0.3.3.dev65__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: eegdash
-Version: 0.3.3.dev61
+Version: 0.3.3.dev65
 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

{eegdash-0.3.3.dev61 → eegdash-0.3.3.dev65}/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.dev65"

{eegdash-0.3.3.dev61 → eegdash-0.3.3.dev65}/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.dev65/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.dev65}/eegdash.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: eegdash
-Version: 0.3.3.dev61
+Version: 0.3.3.dev65
 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

{eegdash-0.3.3.dev61 → eegdash-0.3.3.dev65}/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.dev65/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"