Speed up Client.map by computing token only once for func…

… and ``kwargs`` (#8855)
dask · Sep 3, 2024 · c473680 · c473680
1 parent 2e61816
commit c473680
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/distributed/client.py b/distributed/client.py
@@ -877,8 +877,9 @@ def _keys(self) -> Iterable[Key]:
 
             else:
                 if self.pure:
+                    tok = tokenize(self.func, self.kwargs)
                     keys = [
-                        self.key + "-" + tokenize(self.func, self.kwargs, args)  # type: ignore
+                        self.key + "-" + tokenize(tok, args)  # type: ignore
                         for args in zip(*self.iterables)
                     ]
                 else: