Allow blocking launch of federated tracker. (#10414)

--------- Co-authored-by: Philip Hyunsu Cho <chohyu01@cs.washington.edu>
2024-06-16 01:43:53 +08:00
parent 49e25cfb36
commit 6c83c8c2ef
3 changed files with 18 additions and 4 deletions
--- a/tests/python/test_collective.py
+++ b/tests/python/test_collective.py
@@ -63,7 +63,7 @@ def test_federated_communicator():
    world_size = 2
    tracker = multiprocessing.Process(
        target=federated.run_federated_server,
-        kwargs={"port": port, "n_workers": world_size},
+        kwargs={"port": port, "n_workers": world_size, "blocking": False},
    )
    tracker.start()
    if not tracker.is_alive():