Properly remove worker when note needed

materialsproject · Sep 26, 2022 · 9eda26a · 9eda26a
1 parent b671048
commit 9eda26a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/maggma/cli/distributed.py b/src/maggma/cli/distributed.py
@@ -125,6 +125,7 @@ def manager(
                                     f"Sending exit signal to worker: {msg.split('_')[1]}"
                                 )
                                 socket.send_multipart([identity, b"", b"EXIT"])
+                                workers.pop(identity)
 
                 elif "ERROR" in msg:
                     # Remove worker and requeue work sent to it