Update correct task dict with extractors list

int-brain-lab · Jan 21, 2025 · c924da9 · c924da9
1 parent 8d55feb
commit c924da9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/iblrig_custom_tasks/samuel_cuedBiasedChoiceWorld/task.py b/iblrig_custom_tasks/samuel_cuedBiasedChoiceWorld/task.py
@@ -25,7 +25,7 @@ def __init__(self, *args, **kwargs):
         trials_task = 'CuedBiasedTrials' if is_main_sync else 'CuedBiasedTrialsTimeline'
         self.extractor_tasks = ['TrialRegisterRaw', trials_task, 'TrainingStatus']
         # Update experiment description which was created by superclass init
-        next(iter(self.experiment_description['tasks']))['extractors'] = self.extractor_tasks
+        self.experiment_description['tasks'][-1][self.protocol_name]['extractors'] = self.extractor_tasks
 
         # init behaviour data
         self.movement_left = self.device_rotary_encoder.THRESHOLD_EVENTS[

diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name            = "project_extraction"
-version         = "0.5.4"
+version         = "0.5.5"
 description     = "Custom extractors for satellite tasks"
 dynamic         = [ "readme" ]
 keywords        = [ "IBL", "neuro-science" ]