Fix CI

Signed-off-by: Qing Wang <kingchin1218@gmail.com>
ray-project · Jun 30, 2023 · 2e905e2 · 2e905e2
1 parent 586738a
commit 2e905e2
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/tests/test_ping_others.py b/tests/test_ping_others.py
@@ -17,7 +17,7 @@
 import fed
 import fed._private.compatible_utils as compatible_utils
 import ray
-from fed.barriers import ping_others
+from fed.proxy.barriers import ping_others
 
 
 cluster = {

diff --git a/tests/test_transport_proxy.py b/tests/test_transport_proxy.py
@@ -22,7 +22,7 @@
 import fed._private.compatible_utils as compatible_utils
 from fed._private import constants
 from fed.grpc import fed_pb2, fed_pb2_grpc
-from fed.barriers import send, start_recv_proxy, start_send_proxy
+from fed.proxy.barriers import send, start_recv_proxy, start_send_proxy
 from fed.cleanup import wait_sending
 
 

diff --git a/tests/test_transport_proxy_tls.py b/tests/test_transport_proxy_tls.py
@@ -20,7 +20,7 @@
 
 import fed._private.compatible_utils as compatible_utils
 from fed._private import constants
-from fed.barriers import send, start_recv_proxy, start_send_proxy
+from fed.proxy.barriers import send, start_recv_proxy, start_send_proxy
 from fed.cleanup import wait_sending