fix print

scaleoutsystems · Oct 6, 2023 · deed49e · deed49e
1 parent 7a18790
commit deed49e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/fedn/fedn/network/combiner/round.py b/fedn/fedn/network/combiner/round.py
@@ -204,9 +204,9 @@ def stage_model(self, model_id, timeout_retry=3, retry=2):
 
         # If the model is already in memory at the server we do not need to do anything.
         if self.modelservice.models.exist(model_id):
-            print("MODEL EXISTST (NOT)", flush=True)
+            print("ROUNDCONTROL: Model already exists in memory, skipping model staging.", flush=True)
             return
-        print("MODEL STAGING", flush=True)
+        print("ROUNDCONTROL: Model Staging, fetching model from storage...", flush=True)
         # If not, download it and stage it in memory at the combiner.
         tries = 0
         while True: