Fix typing now that code is tuple of frame(s) (#7778)

dask · Apr 17, 2023 · 5c0a412 · 5c0a412
1 parent 515dffe
commit 5c0a412
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/distributed/scheduler.py b/distributed/scheduler.py
@@ -4260,7 +4260,7 @@ def update_graph(
         user_priority: int | dict[str, int] = 0,
         actors: bool | list[str] | None = None,
         fifo_timeout: float = 0.0,
-        code: str | None = None,
+        code: tuple[str] | None = None,
         annotations: dict | None = None,
         stimulus_id: str | None = None,
     ) -> None: