From 9d816d9988ecd693bd14c66bf2d929ef1a55888c Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Sat, 13 Aug 2022 01:06:52 +0800 Subject: [PATCH] [CI] Test with latest RAPIDS. (#7816) (#8164) --- tests/python-gpu/test_gpu_with_dask.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/python-gpu/test_gpu_with_dask.py b/tests/python-gpu/test_gpu_with_dask.py index 1f0339e91..2e6525f4f 100644 --- a/tests/python-gpu/test_gpu_with_dask.py +++ b/tests/python-gpu/test_gpu_with_dask.py @@ -82,7 +82,7 @@ def run_with_dask_dataframe(DMatrixT: Type, client: Client) -> None: cp.testing.assert_allclose(single_node, predictions.compute()) np.testing.assert_allclose(single_node, - series_predictions.compute().to_array()) + series_predictions.compute().to_numpy()) predt = dxgb.predict(client, out, X) assert isinstance(predt, dd.Series)