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,18 +1,18 @@
 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
 
 from dask import delayed
 
-from distributed import Client, Nanny, Worker, wait, SpecCluster
+from distributed import Client, Nanny, SpecCluster, Worker, wait
 from distributed.chaos import KillWorker
 from distributed.compatibility import WINDOWS
 from distributed.metrics import time
@@ -342,16 +342,16 @@ async def test_chaos_rechunk(c, s, *workers):
 
 @pytest.mark.slow
 def test_stress_scale():
-    for n in range(30):
+    for _ in range(30):
         cluster_kwargs = {}
         client_kwargs = {
-            'set_as_default': False,
+            "set_as_default": False,
         }
         spec = {}
-        template = {'cls': Nanny}
+        template = {"cls": Nanny}
         N = 24
         for i in range(N):
-            w = spec[f'worker-{i}'] = copy.copy(template)
+            w = spec[f"worker-{i}"] = copy.copy(template)
         cluster = SpecCluster(
             workers=spec,
             worker=template,  # <- template for newly scaled up workers