Bug/SK-937 | Update for previous clients fails, missing client_id (#655

)
scaleoutsystems · Jul 11, 2024 · 5af54fc · 5af54fc
1 parent 49e53d2
commit 5af54fc
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/fedn/network/combiner/combiner.py b/fedn/network/combiner/combiner.py
@@ -127,7 +127,10 @@ def __init__(self, config):
         # Set the status to offline for previous clients.
         previous_clients = self.statestore.clients.find({"combiner": config["name"]})
         for client in previous_clients:
-            self.statestore.set_client({"name": client["name"], "status": "offline", "client_id": client["client_id"]})
+            try:
+                self.statestore.set_client({"name": client["name"], "status": "offline", "client_id": client["client_id"]})
+            except KeyError:
+                self.statestore.set_client({"name": client["name"], "status": "offline"})
 
         self.modelservice = ModelService()