Changed to use SiteType.SERVER.

NVIDIA · Nov 26, 2024 · 5af9a2f · 5af9a2f
1 parent f424042
commit 5af9a2f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nvflare/private/fed/simulator/simulator_server.py b/nvflare/private/fed/simulator/simulator_server.py
@@ -15,7 +15,7 @@
 from typing import Dict, List, Optional
 
 from nvflare.apis.fl_component import FLComponent
-from nvflare.apis.fl_constant import FLContextKey, ReservedKey, ReservedTopic, ServerCommandKey
+from nvflare.apis.fl_constant import FLContextKey, ReservedKey, ReservedTopic, ServerCommandKey, SiteType
 from nvflare.apis.fl_context import FLContext
 from nvflare.apis.shareable import ReturnCode, Shareable, make_reply
 from nvflare.apis.workspace import Workspace
@@ -149,7 +149,7 @@ def deploy(self, args, grpc_args=None, secure_train=False):
         os.makedirs(os.path.join(args.workspace, "startup"), exist_ok=True)
         workspace = Workspace(args.workspace, "server", args.config_folder)
         run_manager = RunManager(
-            server_name="server",
+            server_name=SiteType.SERVER,
             engine=self.engine,
             job_id="",
             workspace=workspace,