try to fix linting

DiamondLightSource · Jul 26, 2024 · 3386c7e · 3386c7e
1 parent bf93939
commit 3386c7e
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/dodal/devices/i24/pmac.py b/src/dodal/devices/i24/pmac.py
@@ -120,9 +120,12 @@ def __init__(
         super().__init__(backend, timeout, name)
 
     @AsyncStatus.wrap
-    async def set(self, prog_num: int, timeout: float):
-        prog_str = f"&2b{prog_num}r"
-        await self.signal.set(prog_str, wait=True)
+    async def set(self, value: int, wait=True, timeout=0.0):
+        prog_str = f"&2b{value}r"
+        assert (
+            isinstance(timeout, float) or timeout is None
+        ), "ProgramRunner does not support calculating timeout itself"
+        await self.signal.set(prog_str, wait=wait)
         await wait_for_value(self.status, ScanState.DONE, timeout)