From cf307e450f91c48ab79659143fd6bf6c42366eab Mon Sep 17 00:00:00 2001 From: GALI PREM SAGAR Date: Wed, 30 Jun 2021 18:37:19 -0500 Subject: [PATCH] fix imports (#757) --- debug-tests/client.py | 2 +- tests/test_custom_send_recv.py | 2 +- tests/test_send_recv_two_workers.py | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/debug-tests/client.py b/debug-tests/client.py index d3eb16b28..1e09c8164 100644 --- a/debug-tests/client.py +++ b/debug-tests/client.py @@ -82,7 +82,7 @@ async def read(): # cuda_obj_generator = cloudpickle.loads(func) # pure_cuda_obj = cuda_obj_generator() - # from cudf.tests.utils import assert_eq + # from cudf.testing._utils import assert_eq # import cupy as cp # if isinstance(rx_cuda_obj, cp.ndarray): diff --git a/tests/test_custom_send_recv.py b/tests/test_custom_send_recv.py index 3e1d9532b..9346eaac5 100644 --- a/tests/test_custom_send_recv.py +++ b/tests/test_custom_send_recv.py @@ -115,7 +115,7 @@ async def serve_forever(ep): typ = type(msg) res = typ.deserialize(ucx_header, cudf_buffer) - from cudf.tests.utils import assert_eq + from cudf.testing._utils import assert_eq assert_eq(res, msg) await uu.comm.ep.close() diff --git a/tests/test_send_recv_two_workers.py b/tests/test_send_recv_two_workers.py index f9a327719..705eb782d 100644 --- a/tests/test_send_recv_two_workers.py +++ b/tests/test_send_recv_two_workers.py @@ -12,7 +12,7 @@ from distributed.protocol import to_serialize from distributed.utils import nbytes -import cudf.tests.utils +import cudf.testing._utils import ucp @@ -89,7 +89,7 @@ async def read(): if isinstance(rx_cuda_obj, cupy.ndarray): cupy.testing.assert_allclose(rx_cuda_obj, pure_cuda_obj) else: - cudf.tests.utils.assert_eq(rx_cuda_obj, pure_cuda_obj) + cudf.testing._utils.assert_eq(rx_cuda_obj, pure_cuda_obj) def server(port, func, comm_api):