From 49a5f86e45a831cc64ea3d9cb2e240c67341566e Mon Sep 17 00:00:00 2001 From: Yan Cheng Date: Mon, 14 Oct 2024 14:41:42 -0400 Subject: [PATCH] fix mock executors --- .../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