From 054d814a3de8f0007151cc5789b19bed7a375214 Mon Sep 17 00:00:00 2001 From: James Bourbeau Date: Wed, 3 Jul 2024 16:53:11 -0500 Subject: [PATCH] Fix assert_eq import from cudf --- distributed/comm/tests/test_ucx.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/distributed/comm/tests/test_ucx.py b/distributed/comm/tests/test_ucx.py index f2dd3ecb2e3..fe2e06fc634 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)