Fix CI

ray-project · Jul 21, 2023 · b76feee · b76feee
1 parent 5df87f1
commit b76feee
Showing 1 changed file with 1 addition and 3 deletions.
diff --git 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,
  }},
  )