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 · bf67cd1 · bf67cd1
1 parent 5d22614
commit bf67cd1
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)