Make pre-commit happy

mypy and black mostly

Author: uellue

distributed/deploy/cluster.py

@@ -252,7 +252,7 @@ def _update_worker_status(self, op, msg):
             self.scheduler_info["workers"].update(workers)
             self.scheduler_info.update(msg)
         elif op == "remove":
-            worker = msg['worker']
+            worker = msg["worker"]
             self.scheduler_info["workers"].pop(worker, None)
         else:  # pragma: no cover
             raise ValueError("Invalid op", op, msg)

distributed/diagnostics/plugin.py

@@ -11,7 +11,7 @@
 import tempfile
 import uuid
 import zipfile
-from collections.abc import Awaitable
+from collections.abc import Awaitable, Hashable
 from typing import TYPE_CHECKING, Any, Callable, ClassVar
 
 from dask.typing import Key
@@ -190,7 +190,13 @@ def add_worker(self, scheduler: Scheduler, worker: str) -> None | Awaitable[None
         """
 
     def remove_worker(
-        self, scheduler: Scheduler, worker: str, *, name: str, stimulus_id: str, **kwargs: Any
+        self,
+        scheduler: Scheduler,
+        worker: str,
+        *,
+        name: Hashable,
+        stimulus_id: str,
+        **kwargs: Any,
     ) -> None | Awaitable[None]:
         """Run when a worker leaves the cluster
 

distributed/scheduler.py

@@ -5563,7 +5563,10 @@ async def remove_worker(
             try:
                 try:
                     result = plugin.remove_worker(
-                        scheduler=self, worker=address, name=ws.name, stimulus_id=stimulus_id
+                        scheduler=self,
+                        worker=address,
+                        name=ws.name,
+                        stimulus_id=stimulus_id,
                     )
                 except TypeError:
                     parameters = inspect.signature(plugin.remove_worker).parameters
@@ -9411,8 +9414,8 @@ def add_worker(self, scheduler: Scheduler, worker: str) -> None:
             scheduler.remove_plugin(name=self.name)
 
     def remove_worker(
-            self, scheduler: Scheduler, worker: str, name: str, **kwargs: Any
-        ) -> None:
+        self, scheduler: Scheduler, worker: str, name: Hashable, **kwargs: Any
+    ) -> None:
         try:
             msg = {
                 "worker": worker,