Fix ut.

ray-project · Jan 9, 2024 · 0cd65b0 · 0cd65b0
1 parent f160c31
commit 0cd65b0
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/fed/tests/test_exit_on_failure_sending.py b/fed/tests/test_exit_on_failure_sending.py
@@ -38,8 +38,8 @@ def get_value(self):
 def run(party: str, q: multiprocessing.Queue):
     compatible_utils.init_ray(address='local')
     addresses = {
-        'alice': '127.0.0.1:11012',
-        'bob': '127.0.0.1:11011',
+        'alice': '127.0.0.1:21321',
+        'bob': '127.0.0.1:21322',
     }
     retry_policy = {
         "maxAttempts": 2,
@@ -68,12 +68,6 @@ def failure_handler(error):
     o = f.party("alice").remote()
     My.party("bob").remote(o)
 
-    import time
-
-    # Wait a long time.
-    # If the test takes effect, the main loop here will be broken.
-    time.sleep(86400)
-
 
 def test_exit_when_failure_on_sending():
     q = multiprocessing.Queue()