Worker is already removed by parent function at this point

TODO: Fully understand impact of adding the name parameter on other functions.

Author: uellue

distributed/scheduler.py

@@ -5563,7 +5563,7 @@ async def remove_worker(
             try:
                 try:
                     result = plugin.remove_worker(
-                        scheduler=self, worker=address, stimulus_id=stimulus_id
+                        scheduler=self, worker=address, name=ws.name, stimulus_id=stimulus_id
                     )
                 except TypeError:
                     parameters = inspect.signature(plugin.remove_worker).parameters
@@ -9410,11 +9410,11 @@ 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, **kwargs: Any) -> None:
+    def remove_worker(self, scheduler: Scheduler, worker: str, name: str, **kwargs: Any) -> None:
         try:
             msg = {
                 'worker': worker,
-                'name': scheduler.workers[worker].name,
+                'name': name,
             }
             self.bcomm.send(["remove", msg])
         except CommClosedError: