From 50d854e02e96480ab5bf009db43fd23b35bef7b0 Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Sun, 1 May 2022 02:55:10 +0800 Subject: [PATCH] [CI] Test with latest RAPIDS. (#7816) --- tests/ci_build/Dockerfile.gpu | 2 +- tests/python-gpu/test_gpu_with_dask.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/ci_build/Dockerfile.gpu b/tests/ci_build/Dockerfile.gpu index 72f7b7853..21896bfa0 100644 --- a/tests/ci_build/Dockerfile.gpu +++ b/tests/ci_build/Dockerfile.gpu @@ -19,7 +19,7 @@ ENV PATH=/opt/python/bin:$PATH # Create new Conda environment with cuDF, Dask, and cuPy RUN \ conda create -n gpu_test -c rapidsai-nightly -c rapidsai -c nvidia -c conda-forge -c defaults \ - python=3.8 cudf=21.10* rmm=21.10* cudatoolkit=$CUDA_VERSION_ARG dask dask-cuda=21.10* dask-cudf=21.10* cupy=9.1* \ + python=3.8 cudf=22.04* rmm=22.04* cudatoolkit=$CUDA_VERSION_ARG dask dask-cuda=22.04* dask-cudf=22.04* cupy \ numpy pytest scipy scikit-learn pandas matplotlib wheel python-kubernetes urllib3 graphviz hypothesis ENV GOSU_VERSION 1.10 diff --git a/tests/python-gpu/test_gpu_with_dask.py b/tests/python-gpu/test_gpu_with_dask.py index 2074ce073..6edcbd2f5 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)