diff --git a/moabb/datasets/sosulski2019.py b/moabb/datasets/sosulski2019.py index ade789445..2fc20db7e 100644 --- a/moabb/datasets/sosulski2019.py +++ b/moabb/datasets/sosulski2019.py @@ -7,6 +7,7 @@ from moabb.datasets import download as dl from moabb.datasets.base import BaseDataset +from moabb.datasets.utils import stim_channels_with_selected_ids SPOT_PILOT_P300_URL = ( @@ -95,12 +96,13 @@ def __init__( self.n_channels = 31 self.use_soas_as_sessions = use_soas_as_sessions self.description_map = {"Stimulus/S 21": "Target", "Stimulus/S 1": "NonTarget"} + self.events = dict(Target=21, NonTarget=1) code = "Sosulski2019" interval = [-0.2, 1] if interval is None else interval super().__init__( subjects=list(range(1, 13 + 1)), sessions_per_subject=1, - events=dict(Target=21, NonTarget=1), + events=self.events, code=code, interval=interval, paradigm="p300", @@ -133,7 +135,7 @@ def _get_single_run_data(self, file_path): if self.reject_non_iid: raw.set_annotations(raw.annotations[7:85]) # non-iid rejection raw.annotations.rename(self.description_map) - return raw + return stim_channels_with_selected_ids(raw, self.events) def _get_single_subject_data(self, subject): """Return data for a single subject."""