Hopefully make linters happier

Author: uellue

distributed/deploy/spec.py

@@ -397,12 +397,12 @@ async def _correct_state_internal(self) -> None:
 
     def _update_worker_status(self, op, msg):
         if op == "remove":
-            name = msg['name']
+            name = msg["name"]
 
             def f():
                 if (
                     name in self.workers
-                    and msg['worker'] not in self.scheduler_info["workers"]
+                    and msg["worker"] not in self.scheduler_info["workers"]
                     and not any(
                         d["name"] == name
                         for d in self.scheduler_info["workers"].values()

distributed/diagnostics/plugin.py

@@ -190,7 +190,7 @@ def add_worker(self, scheduler: Scheduler, worker: str) -> None | Awaitable[None
         """
 
     def remove_worker(
-        self, scheduler: Scheduler, worker: str, *, stimulus_id: str, **kwargs: Any
+        self, scheduler: Scheduler, worker: str, name: str, *, stimulus_id: str, **kwargs: Any
     ) -> None | Awaitable[None]:
         """Run when a worker leaves the cluster
 

distributed/scheduler.py

@@ -9410,11 +9410,13 @@ def add_worker(self, scheduler: Scheduler, worker: str) -> None:
         except CommClosedError:
             scheduler.remove_plugin(name=self.name)
 
-    def remove_worker(self, scheduler: Scheduler, worker: str, name: str, **kwargs: Any) -> None:
+    def remove_worker(
+            self, scheduler: Scheduler, worker: str, name: str, **kwargs: Any
+        ) -> None:
         try:
             msg = {
-                'worker': worker,
-                'name': name,
+                "worker": worker,
+                "name": name,
             }
             self.bcomm.send(["remove", msg])
         except CommClosedError:

distributed/tests/test_stress.py

@@ -1,11 +1,11 @@
 from __future__ import annotations
 
 import asyncio
+import copy
 import random
 from contextlib import suppress
 from operator import add
 from time import sleep
-import copy
 
 import pytest
 from tlz import concat, sliding_window