From 4dba2337e85f1912bc431210f81586bb7e726186 Mon Sep 17 00:00:00 2001 From: Olivier Winter Date: Wed, 9 Oct 2024 08:51:36 +0100 Subject: [PATCH] neuromodulatorCW it a subclass of advancedCW (#859) --- ibllib/io/extractors/training_trials.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/ibllib/io/extractors/training_trials.py b/ibllib/io/extractors/training_trials.py index b512d2c37..45195556e 100644 --- a/ibllib/io/extractors/training_trials.py +++ b/ibllib/io/extractors/training_trials.py @@ -127,9 +127,10 @@ def get_trial_repeat(trial): elif 'contrast' in trial and isinstance(trial['contrast'], dict): return trial['contrast']['type'] == 'RepeatContrast' else: - # For advanced choice world before version 8.19.0 there was no 'debias_trial' field + # For advanced choice world and its subclasses before version 8.19.0 there was no 'debias_trial' field # and no debiasing protocol applied, so simply return False - assert self.settings['PYBPOD_PROTOCOL'].startswith('_iblrig_tasks_advancedChoiceWorld') + assert (self.settings['PYBPOD_PROTOCOL'].startswith('_iblrig_tasks_advancedChoiceWorld') or + self.settings['PYBPOD_PROTOCOL'].startswith('ccu_neuromodulatorChoiceWorld')) return False trial_repeated = np.fromiter(map(get_trial_repeat, self.bpod_trials), int)