From 8c799fdaf2e0ca2b183d91306ac63db0a980ae5e Mon Sep 17 00:00:00 2001
From: galipremsagar <sagarprem75@gmail.com>
Date: Wed, 25 May 2022 06:37:49 -0700
Subject: [PATCH 1/3] pin dask for release

---
 ci/gpu/build.sh  | 2 +-
 requirements.txt | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/ci/gpu/build.sh b/ci/gpu/build.sh
index 1363a2156..2e05bb4f0 100755
--- a/ci/gpu/build.sh
+++ b/ci/gpu/build.sh
@@ -35,7 +35,7 @@ export NUMPY_EXPERIMENTAL_ARRAY_FUNCTION=1
 
 # Install dask and distributed from main branch. Usually needed during
 # development time and disabled before a new dask-cuda release.
-export INSTALL_DASK_MAIN=1
+export INSTALL_DASK_MAIN=0
 
 ################################################################################
 # SETUP - Check environment
diff --git a/requirements.txt b/requirements.txt
index 4e8642731..861a4f89b 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
-dask>=2022.03.0
-distributed>=2022.03.0
+dask==2022.05.1
+distributed==2022.05.1
 pynvml>=11.0.0
 numpy>=1.16.0
 numba>=0.54

From be67173e56df3dbf0d582feeb8feae152974053f Mon Sep 17 00:00:00 2001
From: GALI PREM SAGAR <sagarprem75@gmail.com>
Date: Wed, 25 May 2022 09:41:13 -0500
Subject: [PATCH 2/3] Update build.sh

---
 ci/gpu/build.sh | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/ci/gpu/build.sh b/ci/gpu/build.sh
index 2e05bb4f0..286d035d6 100755
--- a/ci/gpu/build.sh
+++ b/ci/gpu/build.sh
@@ -71,8 +71,12 @@ if [[ "${INSTALL_DASK_MAIN}" == 1 ]]; then
   gpuci_mamba_retry install -c dask/label/dev \
     "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
 fi
 
+
 gpuci_logger "Check versions"
 python --version
 $CC --version

From d973e8cbf15e43c29307ba5eeb54fdb9395221e8 Mon Sep 17 00:00:00 2001
From: GALI PREM SAGAR <sagarprem75@gmail.com>
Date: Wed, 25 May 2022 09:53:39 -0500
Subject: [PATCH 3/3] Apply suggestions from code review

Co-authored-by: Peter Andreas Entschev <peter@entschev.com>
---
 ci/gpu/build.sh | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

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