eegdash 0.3.5.dev77__py3-none-any.whl → 0.3.5.dev80__py3-none-any.whl

eegdash/__init__.py

@@ -7,4 +7,4 @@ __init__mongo_client()
 
 __all__ = ["EEGDash", "EEGDashDataset", "EEGChallengeDataset"]
 
-__version__ = "0.3.5.dev77"
+__version__ = "0.3.5.dev80"

eegdash/api.py

@@ -676,6 +676,7 @@ class EEGDashDataset(BaseConcatDataset):
 
         try:
             if records is not None:
+                self.records = records
                 datasets = [
                     EEGDashBaseDataset(
                         record,
@@ -683,16 +684,16 @@ class EEGDashDataset(BaseConcatDataset):
                         self.s3_bucket,
                         **base_dataset_kwargs,
                     )
-                    for record in records
+                    for record in self.records
                 ]
             elif data_dir:
                 # This path loads from a local directory and is not affected by DB query logic
-                if isinstance(data_dir, str):
+                if isinstance(data_dir, str) or isinstance(data_dir, Path):
                     datasets = self.load_bids_dataset(
-                        dataset,
-                        data_dir,
-                        description_fields,
-                        s3_bucket,
+                        dataset=dataset,
+                        data_dir=data_dir,
+                        description_fields=description_fields,
+                        s3_bucket=s3_bucket,
                         **base_dataset_kwargs,
                     )
                 else:
@@ -703,10 +704,10 @@ class EEGDashDataset(BaseConcatDataset):
                     for i, _ in enumerate(data_dir):
                         datasets.extend(
                             self.load_bids_dataset(
-                                dataset[i],
-                                data_dir[i],
-                                description_fields,
-                                s3_bucket,
+                                dataset=dataset[i],
+                                data_dir=data_dir[i],
+                                description_fields=description_fields,
+                                s3_bucket=s3_bucket,
                                 **base_dataset_kwargs,
                             )
                         )
@@ -773,9 +774,9 @@ class EEGDashDataset(BaseConcatDataset):
         """
         datasets: list[EEGDashBaseDataset] = []
 
-        records = self.eeg_dash.find(query, **query_kwargs)
+        self.records = self.eeg_dash.find(query, **query_kwargs)
 
-        for record in records:
+        for record in self.records:
             description = {}
             for field in description_fields:
                 value = self.find_key_in_nested_dict(record, field)
@@ -794,8 +795,8 @@ class EEGDashDataset(BaseConcatDataset):
 
     def load_bids_dataset(
         self,
-        dataset,
-        data_dir,
+        dataset: str,
+        data_dir: str | Path,
         description_fields: list[str],
         s3_bucket: str | None = None,
         **kwargs,

{eegdash-0.3.5.dev77.dist-info → eegdash-0.3.5.dev80.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: eegdash
-Version: 0.3.5.dev77
+Version: 0.3.5.dev80
 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-Expression: GPL-3.0-only

{eegdash-0.3.5.dev77.dist-info → eegdash-0.3.5.dev80.dist-info}/RECORD

@@ -1,5 +1,5 @@
-eegdash/__init__.py,sha256=kEfROPte5G7SgYy3bdM6r3KrWm_XuXcvS-tNPMVBIr4,240
-eegdash/api.py,sha256=M4QX93iqOneqO9zz6ruwGKpmU9h40K8lqYnICqAaj5o,32158
+eegdash/__init__.py,sha256=K-EaG_ZHr-O4aH8SHFg7PP_rbyqlvoa3JcBdlGsXlTU,240
+eegdash/api.py,sha256=KjmEVkfltLR5EwRnmnPp5rEDS5Oa6_dnprif9EVpeQs,32351
 eegdash/data_config.py,sha256=OS6ERO-jHrnEOfMJUehY7ieABdsRw_qWzOKJ4pzSfqw,1323
 eegdash/data_utils.py,sha256=_dycnPmGfTbYs7bc6edHxUn_m01dLYtp92_k44ffEoY,26475
 eegdash/dataset.py,sha256=ooLoxMFy2I8BY9gJl6ncTp_Gz-Rq0Z-o4NJyyomxLcU,2670
@@ -23,8 +23,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.5.dev77.dist-info/licenses/LICENSE,sha256=asisR-xupy_NrQBFXnx6yqXeZcYWLvbAaiETl25iXT0,931
-eegdash-0.3.5.dev77.dist-info/METADATA,sha256=1-0zRsLKzuM36yIDIlpbxTSCbCSp75XQoi2MC0C2pdA,10388
-eegdash-0.3.5.dev77.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-eegdash-0.3.5.dev77.dist-info/top_level.txt,sha256=zavO69HQ6MyZM0aQMR2zUS6TAFc7bnN5GEpDpOpFZzU,8
-eegdash-0.3.5.dev77.dist-info/RECORD,,
+eegdash-0.3.5.dev80.dist-info/licenses/LICENSE,sha256=asisR-xupy_NrQBFXnx6yqXeZcYWLvbAaiETl25iXT0,931
+eegdash-0.3.5.dev80.dist-info/METADATA,sha256=R0-JDW1_w2p1JJjffDbuYSlHJKGv0g7nGmyl3_AtJfY,10388
+eegdash-0.3.5.dev80.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+eegdash-0.3.5.dev80.dist-info/top_level.txt,sha256=zavO69HQ6MyZM0aQMR2zUS6TAFc7bnN5GEpDpOpFZzU,8
+eegdash-0.3.5.dev80.dist-info/RECORD,,