Annotate arguments in new_task

Should knock out type checking when the function is called, which should simplify the work needed in later steps.
dask · Feb 9, 2021 · 63d109e · 63d109e
1 parent f326238
commit 63d109e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/distributed/scheduler.py b/distributed/scheduler.py
@@ -3922,7 +3922,7 @@ def update_graph(
 
         # TODO: balance workers
 
-    def new_task(self, key, spec, state):
+    def new_task(self, key: str, spec: object, state: str):
         """ Create a new task, and associated states """
         parent: SchedulerState = cast(SchedulerState, self)