Adding ERPCore datasets

.pre-commit-config.yaml

@@ -38,7 +38,7 @@ repos:
     rev: 24.3.0
     hooks:
       - id: black
-        language_version: python3.8
+        language_version: python3
         args: [ --line-length=90, --target-version=py38 ]
 
   -   repo: https://github.com/asottile/blacken-docs
@@ -82,7 +82,7 @@ repos:
           - --ignore-words-list=additionals,alle,alot,bund,currenty,datas,farenheit,falsy,fo,haa,hass,iif,incomfort,ines,ist,nam,nd,pres,pullrequests,resset,rime,ser,serie,te,technik,ue,unsecure,withing,zar,crate
           - --skip="./.*,*.csv,*.json,*.ambr"
           - --quiet-level=2
-        exclude_types: [ csv, json, svg ]
+        exclude_types: [ csv, json, svg, pdf ]
         exclude: ^tests/|generated/
 
   - repo: https://github.com/adrienverge/yamllint.git

docs/source/dataset_summary.rst

@@ -65,6 +65,14 @@ P300/ERP
    :class:`Sosulski2019`, 13, 31, 7500 NT / 1500 T, 1.2s, 1000Hz, 1
    :class:`EPFLP300`, 8, 32, 2753 NT / 551 T, 1s, 2048Hz, 4
    :class:`Lee2019_ERP`, 54, 62, 6900 NT / 1380 T, 1s, 1000Hz, 2
+   :class:`ERPCore2021_N170`, 40, 30, 80 , 1s, 1024Hz, 1
+   :class:`ERPCore2021_MMN`, 40, 30, 800 Deviant / 200 Standard, 1s, 1024Hz, 1
+   :class:`ERPCore2021_N2pc`, 40, 30, 160, 1s, 1024Hz, 1
+   :class:`ERPCore2021_P3`, 40, 30, 160 NT / 40 T, 1s, 1024Hz, 1
+   :class:`ERPCore2021_N400`, 40, 30, 60, 1s, 1024Hz, 1
+   :class:`ERPCore2021_ERN`, 40, 30, Depends, 1s, 1024Hz, 1
+   :class:`ERPCore2021_LRP`, 40, 30, Depends, 1s, 1024Hz, 1
+
 
 
 SSVEP

docs/source/whats_new.rst

@@ -17,7 +17,7 @@ Develop branch
 
 Enhancements
 ~~~~~~~~~~~~
-- None
+- Add ERP CORE datasets :class:`moabb.datasets.ErpCore2021` dataset (:gh:`627` by `Taha Habib`_)
 
 Bugs
 ~~~~
@@ -473,3 +473,4 @@ API changes
 .. _Brian Irvine: https://github.com/brianjohannes
 .. _Bruna Lopes: https://github.com/brunaafl
 .. _Yash Chauhan: https://github.com/jiggychauhi
+.. _Taha Habib: https://github.com/tahatt13

moabb/datasets/__init__.py

@@ -56,6 +56,15 @@
     CastillosCVEP100,
 )
 from .epfl import EPFLP300
+from .erpcore2021 import (
+    Erpcore2021_ERN,
+    Erpcore2021_LRP,
+    Erpcore2021_MMN,
+    Erpcore2021_N2pc,
+    Erpcore2021_N170,
+    Erpcore2021_N400,
+    Erpcore2021_P3,
+)
 from .fake import FakeDataset, FakeVirtualRealityDataset
 from .gigadb import Cho2017
 from .hinss2021 import Hinss2021