Update correct task dict with extractors list

Author: k1o0

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[

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" ]