diff --git a/distributed/comm/tests/test_ucx.py b/distributed/comm/tests/test_ucx.py index f2dd3ecb2e..fe2e06fc63 100644 --- 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)