Delete node locally in gRPC-rere (#2596)

adap · Nov 14, 2023 · 04347b2 · 04347b2
1 parent 4f3bb4f
commit 04347b2
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/py/flwr/client/grpc_rere_client/connection.py b/src/py/flwr/client/grpc_rere_client/connection.py
@@ -136,6 +136,8 @@ def delete_node() -> None:
         delete_node_request = DeleteNodeRequest(node=node)
         stub.DeleteNode(request=delete_node_request)
 
+        del node_store[KEY_NODE]
+
     def receive() -> Optional[TaskIns]:
         """Receive next task from server."""
         # Get Node