Fix assert_eq import from cudf (#8747)

dask · Jul 5, 2024 · 86ef1b9 · 86ef1b9
1 parent 670fbfb
commit 86ef1b9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/distributed/comm/tests/test_ucx.py b/distributed/comm/tests/test_ucx.py
@@ -198,7 +198,7 @@ async def test_ping_pong_cudf(ucx_loop, g):
     # *** ImportError: /usr/lib/x86_64-linux-gnu/libstdc++.so.6: version `CXXABI_1.3.11'
     # not found (required by python3.7/site-packages/pyarrow/../../../libarrow.so.12)
     cudf = pytest.importorskip("cudf")
-    from cudf.testing._utils import assert_eq
+    from cudf.testing import assert_eq
 
     cudf_obj = g(cudf)