Fix scheduler tests

dispatchrun · Jun 13, 2024 · 2c2bde3 · 2c2bde3
1 parent ef2bedc
commit 2c2bde3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/dispatch/test_scheduler.py b/tests/dispatch/test_scheduler.py
@@ -449,7 +449,7 @@ def resume(
         poll = self.assert_poll(prev_output)
         input = Input.from_poll_results(
             main.__qualname__,
-            poll.coroutine_state,
+            any_unpickle(poll.typed_coroutine_state),
             call_results,
             Error.from_exception(poll_error) if poll_error else None,
         )