Fix typing issues

pharmbio · Feb 29, 2024 · a56a793 · a56a793
1 parent fee45cf
commit a56a793
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cellpainter/cellpainter/execute.py b/cellpainter/cellpainter/execute.py
@@ -39,7 +39,7 @@ def execute(cmd: Command, runtime: Runtime, metadata: Metadata):
                 execute(c, runtime, metadata)
 
         case Idle():
-            secs = cmd.secs.unwrap()
+            secs = Symbolic.wrap(cmd.secs).unwrap()
             entry = entry.merge(Metadata(est=round(secs, 3)))
             with runtime.timeit(entry):
                 runtime.sleep(secs)
@@ -49,7 +49,7 @@ def execute(cmd: Command, runtime: Runtime, metadata: Metadata):
                 runtime.checkpoint(cmd.name, entry)
 
         case WaitForCheckpoint():
-            plus_secs = cmd.plus_secs.unwrap()
+            plus_secs = Symbolic.wrap(cmd.plus_secs).unwrap()
             t0 = runtime.wait_for_checkpoint(cmd.name)
             desired_point_in_time = t0 + plus_secs
             delay = desired_point_in_time - runtime.monotonic()