From dcd84b3979fccfa62da84e6d60b57969f7af588f Mon Sep 17 00:00:00 2001 From: Jiaming Yuan Date: Fri, 18 Jun 2021 10:27:51 +0800 Subject: [PATCH] [CI] Configure RAPIDS, dask, modin (#7033) Co-authored-by: Philip Hyunsu Cho --- tests/ci_build/Dockerfile.gpu | 2 +- tests/ci_build/Dockerfile.rmm | 2 +- tests/ci_build/conda_env/macos_cpu_test.yml | 4 ++-- tests/ci_build/test_python.sh | 1 + 4 files changed, 5 insertions(+), 4 deletions(-) diff --git a/tests/ci_build/Dockerfile.gpu b/tests/ci_build/Dockerfile.gpu index add4efbf2fd3..386d2f609751 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.7 cudf=21.06* rmm=21.06* cudatoolkit=$CUDA_VERSION_ARG dask=2021.05.0 dask-cuda dask-cudf cupy \ + python=3.7 cudf=21.08* rmm=21.08* cudatoolkit=$CUDA_VERSION_ARG dask dask-cuda dask-cudf cupy \ numpy pytest scipy scikit-learn pandas matplotlib wheel python-kubernetes urllib3 graphviz hypothesis ENV GOSU_VERSION 1.10 diff --git a/tests/ci_build/Dockerfile.rmm b/tests/ci_build/Dockerfile.rmm index 2258925c3815..a7e00f9c1262 100644 --- a/tests/ci_build/Dockerfile.rmm +++ b/tests/ci_build/Dockerfile.rmm @@ -29,7 +29,7 @@ ENV PATH=/opt/python/bin:$PATH # Create new Conda environment with RMM RUN \ conda create -n gpu_test -c rapidsai-nightly -c rapidsai -c nvidia -c conda-forge -c defaults \ - python=3.7 rmm=21.06* cudatoolkit=$CUDA_VERSION_ARG + python=3.7 rmm=21.08* cudatoolkit=$CUDA_VERSION_ARG ENV GOSU_VERSION 1.10 diff --git a/tests/ci_build/conda_env/macos_cpu_test.yml b/tests/ci_build/conda_env/macos_cpu_test.yml index 3180053146de..6a9092b2bbf0 100644 --- a/tests/ci_build/conda_env/macos_cpu_test.yml +++ b/tests/ci_build/conda_env/macos_cpu_test.yml @@ -13,8 +13,8 @@ dependencies: - scikit-learn - pandas - matplotlib -- dask=2021.05.0 -- distributed=2021.05.0 +- dask +- distributed - graphviz - python-graphviz - hypothesis diff --git a/tests/ci_build/test_python.sh b/tests/ci_build/test_python.sh index e07d545038c0..8ecfb766126a 100755 --- a/tests/ci_build/test_python.sh +++ b/tests/ci_build/test_python.sh @@ -60,6 +60,7 @@ case "$suite" in cpu) source activate cpu_test install_xgboost + export RAY_OBJECT_STORE_ALLOW_SLOW_STORAGE=1 pytest -v -s -rxXs --fulltrace --durations=0 ${args} tests/python cd tests/distributed ./runtests.sh