Skip to content

Commit

Permalink
Merge branch 'develop' into n_subjects
Browse files Browse the repository at this point in the history
  • Loading branch information
bruAristimunha authored Oct 29, 2023
2 parents 21be5ab + f1cdc6b commit 35af427
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 1 deletion.
2 changes: 1 addition & 1 deletion docs/source/whats_new.rst
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ Develop branch
Enhancements
~~~~~~~~~~~~

- None
- Adding cache option to the evaluation (:gh:`517` by `Bruno Aristimunha`_)

Bugs
~~~~
Expand Down
2 changes: 2 additions & 0 deletions moabb/evaluations/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -67,6 +67,7 @@ def __init__(
mne_labels=False,
n_splits=None,
save_model=False,
cache_config=None,
):
self.random_state = random_state
self.n_jobs = n_jobs
Expand All @@ -78,6 +79,7 @@ def __init__(
self.mne_labels = mne_labels
self.n_splits = n_splits
self.save_model = save_model
self.cache_config = cache_config
# check paradigm
if not isinstance(paradigm, BaseParadigm):
raise (ValueError("paradigm must be an Paradigm instance"))
Expand Down
3 changes: 3 additions & 0 deletions moabb/evaluations/evaluations.py
Original file line number Diff line number Diff line change
Expand Up @@ -162,6 +162,7 @@ def _evaluate(
subjects=[subject],
return_epochs=self.return_epochs,
return_raws=self.return_raws,
cache_config=self.cache_config,
postprocess_pipeline=postprocess_pipeline,
)
# iterate over sessions
Expand Down Expand Up @@ -491,6 +492,7 @@ def evaluate(
subjects=[subject],
return_epochs=self.return_epochs,
return_raws=self.return_raws,
cache_config=self.cache_config,
postprocess_pipeline=postprocess_pipeline,
)
le = LabelEncoder()
Expand Down Expand Up @@ -661,6 +663,7 @@ def evaluate(
dataset=dataset,
return_epochs=self.return_epochs,
return_raws=self.return_raws,
cache_config=self.cache_config,
postprocess_pipeline=postprocess_pipeline,
)

Expand Down

0 comments on commit 35af427

Please sign in to comment.