From b76feeeb3529419cd54b97af24ce69ab2cd16a0d Mon Sep 17 00:00:00 2001 From: Qing Wang Date: Fri, 21 Jul 2023 13:43:37 +0800 Subject: [PATCH] Fix CI --- tests/test_retry_policy.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/tests/test_retry_policy.py b/tests/test_retry_policy.py index d0407ab..92981aa 100644 --- a/tests/test_retry_policy.py +++ b/tests/test_retry_policy.py @@ -20,8 +20,6 @@ import fed._private.compatible_utils as compatible_utils import ray -from fed.config import GrpcCrossSiloMessageConfig - @fed.remote def f(): @@ -54,7 +52,7 @@ def run(party, is_inner_party): addresses=addresses, party=party, config={'cross_silo_message': { - 'grpc_retry_policy': retry_policy, + 'grpc_retry_policy': retry_policy, }}, )