From 8a1962a5935e639fefb0c2aebc97be3c9a4ec71c Mon Sep 17 00:00:00 2001 From: Peter Andreas Entschev Date: Wed, 26 Jun 2024 00:42:20 -0700 Subject: [PATCH] Update cuDF's `assert_eq` import https://github.com/rapidsai/cudf/pull/16063 has updated the import location of `assert_eq` to the public `cudf.testing.assert_eq`, this change updates imports accordingly. --- dask_cuda/tests/test_cudf_builtin_spilling.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dask_cuda/tests/test_cudf_builtin_spilling.py b/dask_cuda/tests/test_cudf_builtin_spilling.py index d4c28ba06..80b1d482d 100644 --- a/dask_cuda/tests/test_cudf_builtin_spilling.py +++ b/dask_cuda/tests/test_cudf_builtin_spilling.py @@ -20,7 +20,7 @@ get_global_manager, set_global_manager, ) -from cudf.testing._utils import assert_eq # noqa: E402 +from cudf.testing import assert_eq # noqa: E402 if get_global_manager() is not None: pytest.skip(