Skip to content

Commit

Permalink
[FIX] Solving Liu dataset!
Browse files Browse the repository at this point in the history
  • Loading branch information
bruAristimunha committed Oct 23, 2024
1 parent 74895cb commit a24b97c
Showing 1 changed file with 6 additions and 5 deletions.
11 changes: 6 additions & 5 deletions moabb/datasets/liu2024.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,6 +14,7 @@

from moabb.datasets import download as dl
from moabb.datasets.base import BaseDataset
from moabb.datasets.utils import stim_channels_with_selected_ids


# Link to the raw data
Expand Down Expand Up @@ -77,15 +78,15 @@ class Liu2024(BaseDataset):
def __init__(self, break_events=False, instr_events=False):
self.break_events = break_events
self.instr_events = instr_events
events = {"left_hand": 1, "right_hand": 2}
self.events = {"left_hand": 1, "right_hand": 2}
if break_events:
events["instr"] = 3
self.events["instr"] = 3
if instr_events:
events["break"] = 4
self.events["break"] = 4
super().__init__(
subjects=list(range(1, 50 + 1)),
sessions_per_subject=1,
events=events,
events=self.events,
code="Liu2024",
interval=(2, 6),
paradigm="imagery",
Expand Down Expand Up @@ -277,7 +278,7 @@ def _get_single_subject_data(self, subject):
# Loading dataset
raw = raw.load_data(verbose=False)
# There is only one session
sessions = {"0": {"0": raw}}
sessions = {"0": {"0": stim_channels_with_selected_ids(raw, self.event_id)}}

return sessions

Expand Down

0 comments on commit a24b97c

Please sign in to comment.