reconfigure setup

Author: dungscout96

pyproject.toml

@@ -29,6 +29,11 @@ dependencies = [
   "joblib",
   "braindecode",
   "mne-bids",
+  "pybids",
+  "pymatreader",
+  "pyarrow",
+  "tqdm",
+  "numba",
 ]
 [project.urls]
 Homepage = "https://github.com/sccn/EEG-Dash-Data"

src/eegdash/config.json renamed to src/eegdash/config_old.json

@@ -0,0 +1,28 @@
+config = {
+  "required_fields": ["data_name"],
+  "attributes": {
+    "data_name": "str",
+    "dataset": "str",
+    "bidspath": "str",
+    "subject": "str",
+    "task": "str",
+    "session": "str",
+    "run": "str",
+    "sampling_frequency": "float",
+    "modality": "str",
+    "nchans": "int",
+    "ntimes": "int"
+  },
+  "description_fields": ["subject", "session", "run", "task", "age", "gender", "sex"],
+  "bids_dependencies_files": [
+    "dataset_description.json", 
+    "participants.tsv", 
+    "events.tsv", 
+    "events.json", 
+    "eeg.json", 
+    "electrodes.tsv", 
+    "channels.tsv", 
+    "coordsystem.json"
+  ],
+  "accepted_query_fields": ["data_name", "dataset"]
+}

src/eegdash/data_config.py

@@ -10,6 +10,7 @@
 import numpy as np
 import xarray as xr
 from .data_utils import EEGBIDSDataset, EEGDashBaseRaw, EEGDashBaseDataset
+from .data_config import config as data_config
 from braindecode.datasets import BaseDataset, BaseConcatDataset
 from collections import defaultdict
 from pymongo import MongoClient, InsertOne, UpdateOne, DeleteOne
@@ -19,10 +20,11 @@ class EEGDash:
     def __init__(self, 
                  is_public=True):
         # Load config file
-        config_path = Path(__file__).parent / 'config.json'
-        with open(config_path, 'r') as f:
-            self.config = json.load(f)
+        # config_path = Path(__file__).parent / 'config.json'
+        # with open(config_path, 'r') as f:
+        #     self.config = json.load(f)
 
+        self.config = data_config
         if is_public:
             DB_CONNECTION_STRING="mongodb+srv://eegdash-user:[email protected]/?retryWrites=true&w=majority&appName=Cluster0"
         else: