add some details

Author: bimac

iblrig_custom_tasks/nate_adaptiveTimeoutChoiceWorld/task.py

@@ -21,7 +21,24 @@
 
 
 class AdaptiveTimeoutStateMachine(StateMachine):
+
+    def __init__(
+        self,
+        bpod,
+        adaptive_delay_nogo,
+        adaptive_delay_error
+    ):
+        super().__init__(bpod)
+        self.adaptive_delay_nogo = adaptive_delay_nogo
+        self.adaptive_delay_error = adaptive_delay_error
+
+
     def add_state(self, **kwargs):
+        match kwargs['state_name']:
+            case 'nogo':
+                pass
+            case 'error':
+                pass
         super().add_state(**kwargs)
 
 
@@ -38,9 +55,11 @@ def __init__(
         self.adaptive_delay_nogo = adaptive_delay_nogo
         self.adaptive_delay_error = adaptive_delay_error
         super().__init__(*args, **kwargs)
+        assert len(self.adaptive_delay_nogo) == len(self.task_params.CONTRAST_SET)
+        assert len(self.adaptive_delay_error) == len(self.task_params.CONTRAST_SET)
 
     def _instantiate_state_machine(self, trial_number=None):
-        return AdaptiveTimeoutStateMachine(self.bpod)
+        return AdaptiveTimeoutStateMachine(self.bpod, self.adaptive_delay_nogo, self.adaptive_delay_error)
 
     @staticmethod
     def extra_parser():

iblrig_custom_tasks/nate_adaptiveTimeoutChoiceWorld/task_parameters.yaml

@@ -1,3 +1,10 @@
-# define delays for the following contrasts: [1.0, 0.25, 0.125, 0.0625, 0.0]
-'ADAPTIVE_FEEDBACK_NOGO_DELAY_SECS': [5.0, 4.0, 3.0, 2.0, 2.0]
-'ADAPTIVE_FEEDBACK_ERROR_DELAY_SECS': [5.0, 4.0, 3.0, 2.0, 2.0]
+'ADAPTIVE_GAIN': True
+'ADAPTIVE_REWARD': True
+'AG_INIT_VALUE': 8.0  # Adaptive Gain init value. Once the mouse completes 200 response trials whithin a session, this reverts to STIM_GAIN
+'CONTRAST_SET_PROBABILITY_TYPE': skew_zero  # uniform, skew_zero
+'DEBIAS': True  # Whether to use debiasing rule or not by repeating error trials
+'REWARD_AMOUNT_UL': 3.0  # Reward amount (uL), will oscillate between 1.5 and 3 uL depending on previous sessions if adaptive_reward is True
+
+'CONTRAST_SET': [1.0, 0.5, 0.25, 0.125, 0.0625, 0.0]
+'ADAPTIVE_FEEDBACK_NOGO_DELAY_SECS': [6.0, 5.0, 4.0, 3.0, 2.0, 2.0]
+'ADAPTIVE_FEEDBACK_ERROR_DELAY_SECS': [6.0, 5.0, 4.0, 3.0, 2.0, 2.0]