fix imports (#757)

rapidsai · Jun 30, 2021 · cf307e4 · cf307e4
1 parent 27f39c7
commit cf307e4
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -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
@@ -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):