FIX CI

Signed-off-by: Qing Wang <kingchin1218@gmail.com>
ray-project · Jul 6, 2023 · 304ff19 · 304ff19
1 parent 85da72b
commit 304ff19
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 3 deletions.
diff --git a/fed/proxy/barriers.py b/fed/proxy/barriers.py
@@ -469,7 +469,6 @@ def send(
         downstream_seq_id=downstream_seq_id,
     )
     get_global_context()._cleanup_manager.push_to_sending(res)
-    # push_to_sending(res)
     return res
 
 

diff --git a/tests/test_transport_proxy.py b/tests/test_transport_proxy.py
@@ -70,7 +70,6 @@ def test_n_to_1_transport():
     for i in range(NUM_DATA):
         assert f"data-{i}" in ray.get(get_objs)
 
-    wait_sending()
     ray.shutdown()
 
 
@@ -186,7 +185,6 @@ def test_send_grpc_with_meta():
     for result in ray.get(sent_objs):
         assert result
 
-    wait_sending()
     ray.shutdown()