From 019ea901ef4daaf8f904a0e776df05e8af8b21d5 Mon Sep 17 00:00:00 2001
From: GALI PREM SAGAR <sagarprem75@gmail.com>
Date: Thu, 26 May 2022 04:10:52 -0500
Subject: [PATCH] Pin `dask` & `distributed` for release (#916)

Pinnings to be in-line with: https://github.com/rapidsai/cudf/pull/10965

Authors:
  - GALI PREM SAGAR (https://github.com/galipremsagar)

Approvers:
  - Ray Douglass (https://github.com/raydouglass)

URL: https://github.com/rapidsai/dask-cuda/pull/916
---
 ci/gpu/build.sh  | 9 ++++++++-
 requirements.txt | 4 ++--
 2 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/ci/gpu/build.sh b/ci/gpu/build.sh
index 1363a2156..5a2d5111c 100755
--- a/ci/gpu/build.sh
+++ b/ci/gpu/build.sh
@@ -35,7 +35,10 @@ 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
+
+# Dask version to install when `INSTALL_DASK_MAIN=0`
+export DASK_STABLE_VERSION="2022.05.1"
 
 ################################################################################
 # SETUP - Check environment
@@ -71,8 +74,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==${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
 
+
 gpuci_logger "Check versions"
 python --version
 $CC --version
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