From efec9bfe6b15ff041055ebdf0f5aba80a3324b27 Mon Sep 17 00:00:00 2001
From: Yan Cheng <58191769+yanchengnv@users.noreply.github.com>
Date: Mon, 14 Oct 2024 15:02:41 -0400
Subject: [PATCH] fix mock executors (#3040)

---
 .../xgboost/histogram_based_v2/mock/mock_executor.py        | 6 ++++--
 .../xgboost/histogram_based_v2/mock/mock_secure_executor.py | 6 ++++--
 2 files changed, 8 insertions(+), 4 deletions(-)

diff --git a/nvflare/app_opt/xgboost/histogram_based_v2/mock/mock_executor.py b/nvflare/app_opt/xgboost/histogram_based_v2/mock/mock_executor.py
index 6226297b24..669389b1d5 100644
--- a/nvflare/app_opt/xgboost/histogram_based_v2/mock/mock_executor.py
+++ b/nvflare/app_opt/xgboost/histogram_based_v2/mock/mock_executor.py
@@ -21,13 +21,15 @@ class MockXGBExecutor(XGBExecutor):
     def __init__(
         self,
         int_server_grpc_options=None,
-        req_timeout=10.0,
+        per_msg_timeout=10.0,
+        tx_timeout=60.0,
         in_process=True,
     ):
         XGBExecutor.__init__(
             self,
             adaptor_component_id="",
-            req_timeout=req_timeout,
+            per_msg_timeout=per_msg_timeout,
+            tx_timeout=tx_timeout,
         )
         self.int_server_grpc_options = int_server_grpc_options
         self.in_process = in_process
diff --git a/nvflare/app_opt/xgboost/histogram_based_v2/mock/mock_secure_executor.py b/nvflare/app_opt/xgboost/histogram_based_v2/mock/mock_secure_executor.py
index e4c4f3e23c..d4a24684d4 100644
--- a/nvflare/app_opt/xgboost/histogram_based_v2/mock/mock_secure_executor.py
+++ b/nvflare/app_opt/xgboost/histogram_based_v2/mock/mock_secure_executor.py
@@ -21,13 +21,15 @@ class MockSecureXGBExecutor(XGBExecutor):
     def __init__(
         self,
         int_server_grpc_options=None,
-        req_timeout=10.0,
+        per_msg_timeout=10.0,
+        tx_timeout=100.0,
         in_process=True,
     ):
         XGBExecutor.__init__(
             self,
             adaptor_component_id="",
-            req_timeout=req_timeout,
+            per_msg_timeout=per_msg_timeout,
+            tx_timeout=tx_timeout,
         )
         self.int_server_grpc_options = int_server_grpc_options
         self.in_process = in_process