From 2e905e2a472683fa1d62b4b453d55d654ec9d20d Mon Sep 17 00:00:00 2001 From: Qing Wang Date: Fri, 30 Jun 2023 16:50:06 +0800 Subject: [PATCH] Fix CI Signed-off-by: Qing Wang --- tests/test_ping_others.py | 2 +- tests/test_transport_proxy.py | 2 +- tests/test_transport_proxy_tls.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_ping_others.py b/tests/test_ping_others.py index 71d99d9..1f9c6b3 100644 --- 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 index 2f8d19c..89a3ecd 100644 --- 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 index fdfe7e3..0fe5d95 100644 --- 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