diff --git a/ci/gpu/build.sh b/ci/gpu/build.sh index 286d035d6..5a2d5111c 100755 --- a/ci/gpu/build.sh +++ b/ci/gpu/build.sh @@ -37,6 +37,9 @@ export NUMPY_EXPERIMENTAL_ARRAY_FUNCTION=1 # development time and disabled before a new dask-cuda release. export INSTALL_DASK_MAIN=0 +# Dask version to install when `INSTALL_DASK_MAIN=0` +export DASK_STABLE_VERSION="2022.05.1" + ################################################################################ # SETUP - Check environment ################################################################################ @@ -72,8 +75,8 @@ if [[ "${INSTALL_DASK_MAIN}" == 1 ]]; then "dask/label/dev::dask" \ "dask/label/dev::distributed" else - gpuci_logger "gpuci_mamba_retry install conda-forge::dask==2022.05.1 conda-forge::distributed==2022.05.1 conda-forge::dask-core==2022.05.1 --force-reinstall" - gpuci_mamba_retry install conda-forge::dask==2022.05.1 conda-forge::distributed==2022.05.1 conda-forge::dask-core==2022.05.1 --force-reinstall + gpuci_logger "gpuci_mamba_retry install conda-forge::dask==${DASK_STABLE_VERSION} conda-forge::distributed==${DASK_STABLE_VERSION} conda-forge::dask-core==${DASK_STABLE_VERSION} --force-reinstall" + gpuci_mamba_retry install conda-forge::dask==${DASK_STABLE_VERSION} conda-forge::distributed==${DASK_STABLE_VERSION} conda-forge::dask-core==${DASK_STABLE_VERSION} --force-reinstall fi