fix regressions

dask · Sep 27, 2021 · 2728850 · 2728850
1 parent 85dc2ed
commit 2728850
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/distributed/_ipython_utils.py b/distributed/_ipython_utils.py
@@ -130,7 +130,7 @@ def remote_magic(line, cell=None):
 
 
 # cache clients for re-use in remote magic
-remote_magic._clients: dict[str, BlockingKernelClient] = {}
+remote_magic._clients = {}  # type: ignore
 
 
 def register_remote_magic(magic_name="remote"):

diff --git a/distributed/worker.py b/distributed/worker.py
@@ -2585,7 +2585,7 @@ async def gather_dep(
                         ("busy-gather", worker, to_gather_keys, stimulus_id, time())
                     )
 
-                recommendations = {}
+                recommendations: dict[TaskState, str | tuple] = {}
                 deps_to_iter = self.in_flight_workers.pop(worker)
 
                 for d in deps_to_iter:

diff --git a/setup.py b/setup.py
@@ -43,8 +43,8 @@
     except ImportError:
         setup_requires.append("cython")
 
-    _, _, params = cython_arg.partition("=")
-    params = params.split(",")
+    _, _, params_str = cython_arg.partition("=")
+    params = params_str.split(",")
     profile = "profile" in params
     if "annotate" in params:
         import Cython.Compiler.Options