default val for Grpc config

ray-project · Jul 10, 2023 · cdc1996 · cdc1996
1 parent 582f523
commit cdc1996
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/tests/test_exit_on_failure_sending.py b/tests/test_exit_on_failure_sending.py
@@ -19,7 +19,7 @@
 import fed
 import fed._private.compatible_utils as compatible_utils
 
-from fed.config import CrossSiloCommConfig
+from fed.config import CrossSiloGrpcCommConfig
 
 import signal
 
@@ -63,7 +63,7 @@ def run(party, is_inner_party):
         "backoffMultiplier": 1,
         "retryableStatusCodes": ["UNAVAILABLE"],
     }
-    cross_silo_comm_config = CrossSiloCommConfig(
+    cross_silo_comm_config = CrossSiloGrpcCommConfig(
         grpc_retry_policy=retry_policy,
         exit_on_sending_failure=True
     )
@@ -90,4 +90,5 @@ def test_exit_when_failure_on_sending():
 
 
 if __name__ == "__main__":
-    sys.exit(pytest.main(["-sv", __file__]))
+    # sys.exit(pytest.main(["-sv", __file__]))
+    test_exit_when_failure_on_sending()