From 0efc560e75273892c891f96740a4e752b4c50918 Mon Sep 17 00:00:00 2001
From: galipremsagar <sagarprem75@gmail.com>
Date: Tue, 1 Feb 2022 12:42:28 -0800
Subject: [PATCH] modify comments

---
 ci/benchmark/build.sh | 8 ++++----
 ci/gpu/build.sh       | 8 +++++---
 2 files changed, 9 insertions(+), 7 deletions(-)

diff --git a/ci/benchmark/build.sh b/ci/benchmark/build.sh
index 3095695c01d..c7e15d9c5f6 100755
--- a/ci/benchmark/build.sh
+++ b/ci/benchmark/build.sh
@@ -36,7 +36,7 @@ export GBENCH_BENCHMARKS_DIR="$WORKSPACE/cpp/build/gbenchmarks/"
 # like `/tmp` is.
 export LIBCUDF_KERNEL_CACHE_PATH="$HOME/.jitify-cache"
 
-# Dask & Distributed option to install main
+# Dask & Distributed option to install main(nightly) or `conda-forge` packages.
 export INSTALL_DASK_MAIN=1
 
 function remove_libcudf_kernel_cache_dir {
@@ -77,16 +77,16 @@ conda install "rmm=$MINOR_VERSION.*" "cudatoolkit=$CUDA_REL" \
 # conda remove -f rapids-build-env rapids-notebook-env
 # conda install "your-pkg=1.0.0"
 
-# Install the master version of dask, distributed, and streamz
-# Install the main version of dask and distributed
+# Install the conda-forge or nightly version of dask and distributed
 if [[ "${INSTALL_DASK_MAIN}" == 1 ]]; then
     gpuci_logger "conda install -c dask/label/dev dask"
     conda install -c dask/label/dev dask
 else
-    gpuci_logger "conda install -c conda-forge dask"
+    gpuci_logger "conda install -c conda-forge dask>=2021.11.1"
     conda install -c conda-forge dask
 fi
 
+# Install the master version of streamz
 logger "pip install git+https://github.com/python-streamz/streamz.git@master --upgrade --no-deps"
 pip install "git+https://github.com/python-streamz/streamz.git@master" --upgrade --no-deps
 
diff --git a/ci/gpu/build.sh b/ci/gpu/build.sh
index a5fa9e03b1f..4fb90cca6f7 100755
--- a/ci/gpu/build.sh
+++ b/ci/gpu/build.sh
@@ -30,7 +30,7 @@ export CONDA_ARTIFACT_PATH="$WORKSPACE/ci/artifacts/cudf/cpu/.conda-bld/"
 export GIT_DESCRIBE_TAG=`git describe --tags`
 export MINOR_VERSION=`echo $GIT_DESCRIBE_TAG | grep -o -E '([0-9]+\.[0-9]+)'`
 
-# Dask & Distributed git tag
+# Dask & Distributed option to install main(nightly) or `conda-forge` packages.
 export INSTALL_DASK_MAIN=1
 
 # ucx-py version
@@ -104,8 +104,8 @@ conda config --show-sources
 conda list --show-channel-urls
 
 function install_dask {
-    # Install the main version of dask, distributed, and streamz
-    gpuci_logger "Install the main version of dask, distributed, and streamz"
+    # Install the conda-forge or nightly version of dask and distributed
+    gpuci_logger "Install the conda-forge or nightly version of dask and distributed"
     set -x
     if [[ "${INSTALL_DASK_MAIN}" == 1 ]]; then
         gpuci_logger "conda install -c dask/label/dev dask"
@@ -114,6 +114,8 @@ function install_dask {
         gpuci_logger "conda install -c conda-forge dask"
         conda install -c conda-forge dask
     fi
+    # Install the main version of streamz
+    gpuci_logger "Install the main version of streamz"
     # Need to uninstall streamz that is already in the env.
     pip uninstall -y streamz
     pip install "git+https://github.com/python-streamz/streamz.git@master" --upgrade --no-deps