Make type-checking happy

DiamondLightSource · Oct 24, 2024 · 7827da8 · 7827da8
1 parent 3793660
commit 7827da8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -190,11 +190,11 @@ def stop_event_loop():
     del RE
 
 
-def pass_on_mock(motor, call_log: MagicMock | None = None):
-    def _pass_on_mock(value, **kwargs):
+def pass_on_mock(motor: Motor, call_log: MagicMock | None = None):
+    def _pass_on_mock(value: float, wait: bool):
         set_mock_value(motor.user_readback, value)
         if call_log is not None:
-            call_log(value, **kwargs)
+            call_log(value, wait=wait)
 
     return _pass_on_mock