fix CI

int-brain-lab · Oct 18, 2024 · 9801a3e · 9801a3e
1 parent f53a195
commit 9801a3e
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 7 deletions.
diff --git a/iblrig/base_choice_world.py b/iblrig/base_choice_world.py
@@ -479,10 +479,6 @@ def next_trial(self):
     def default_reward_amount(self):
         return self.task_params.REWARD_AMOUNT_UL
 
-    @property
-    def stimulus_gain(self) -> float:
-        return self.task_params.STIM_GAIN
-
     def draw_next_trial_info(self, pleft=0.5, **kwargs):
         """Draw next trial variables.
 

diff --git a/iblrig/base_tasks.py b/iblrig/base_tasks.py
@@ -1001,7 +1001,7 @@ class RotaryEncoderMixin(BaseSession, HasBpod):
 
     @property
     def stimulus_gain(self) -> float:
-        return self.task_params.STIM_GAIN()
+        return self.task_params.STIM_GAIN
 
     def init_mixin_rotary_encoder(self):
         thresholds_deg = self.task_params.STIM_POSITIONS + self.task_params.QUIESCENCE_THRESHOLDS

diff --git a/iblrig/test/test_hardware_mixins.py b/iblrig/test/test_hardware_mixins.py
@@ -101,8 +101,8 @@ def test_rotary_encoder_mixin(self):
         """
         Instantiates a bare session with the rotary encoder mixin
         """
-        session = self.session
-        RotaryEncoderMixin.init_mixin_rotary_encoder(session)
+        RotaryEncoderSession = type('RotaryEncoderSession', (EmptyHardwareSession, RotaryEncoderMixin), {})  # noqa: N806
+        session = RotaryEncoderSession(task_parameter_file=ChoiceWorldSession.base_parameters_file, **TASK_KWARGS)
         assert session.device_rotary_encoder.ENCODER_EVENTS == [
             'RotaryEncoder1_1',
             'RotaryEncoder1_2',