Skip to content

Commit

Permalink
Easy fix with Sosulski2019 ;)
Browse files Browse the repository at this point in the history
  • Loading branch information
bruAristimunha committed Oct 23, 2024
1 parent ca0766b commit 1c927ea
Showing 1 changed file with 4 additions and 2 deletions.
6 changes: 4 additions & 2 deletions moabb/datasets/sosulski2019.py
Original file line number Diff line number Diff line change
Expand Up @@ -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 = (
Expand Down Expand Up @@ -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",
Expand Down Expand Up @@ -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."""
Expand Down

0 comments on commit 1c927ea

Please sign in to comment.