Fix xgboost model overwrite (#2663)

NVIDIA · Jun 25, 2024 · e63c4f6 · e63c4f6
1 parent 4f604bb
commit e63c4f6
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/nvflare/app_opt/xgboost/histogram_based_v2/adaptors/grpc_client_adaptor.py b/nvflare/app_opt/xgboost/histogram_based_v2/adaptors/grpc_client_adaptor.py
@@ -67,6 +67,8 @@ class since the self object contains a sender that contains a Core Cell which ca
         Returns: None
 
         """
+        simulate_mode = fl_ctx.get_prop(FLContextKey.SIMULATE_MODE, False)
+        model_dir = self._app_dir if simulate_mode else self._run_dir
         runner_ctx = {
             Constant.RUNNER_CTX_WORLD_SIZE: self.world_size,
             Constant.RUNNER_CTX_CLIENT_NAME: self._client_name,
@@ -76,7 +78,7 @@ class since the self object contains a sender that contains a Core Cell which ca
             Constant.RUNNER_CTX_TRAINING_MODE: self.training_mode,
             Constant.RUNNER_CTX_XGB_PARAMS: self.xgb_params,
             Constant.RUNNER_CTX_XGB_OPTIONS: self.xgb_options,
-            Constant.RUNNER_CTX_MODEL_DIR: self._run_dir,
+            Constant.RUNNER_CTX_MODEL_DIR: model_dir,
             Constant.RUNNER_CTX_TB_DIR: self._app_dir,
         }
         self.start_runner(runner_ctx, fl_ctx)