Pin dask and distributed for release (rapidsai#4999)

This PR pins `dask` and `distributed` to `2022.11.1` for `22.12` release. xref: rapidsai/cudf#12165 Authors: - GALI PREM SAGAR (https://github.com/galipremsagar) Approvers: - Ray Douglass (https://github.com/raydouglass) URL: rapidsai#4999
jakirkham · Dec 1, 2022 · f05be2e · f05be2e
1 parent debdd12
commit f05be2e
Show file tree

Hide file tree

Showing 7 changed files with 15 additions and 15 deletions.
diff --git a/.github/workflows/wheels.yml b/.github/workflows/wheels.yml
@@ -53,10 +53,10 @@ jobs:
       skbuild-configure-options: "-DCUML_BUILD_WHEELS=ON -DDETECT_CONDA_ENV=OFF -DCPM_cumlprims_mg_SOURCE=/project/cumlprims_mg/"
 
       # Always want to test against latest dask/distributed.
-      test-before-amd64: "pip install git+https://github.com/dask/dask.git@main git+https://github.com/dask/distributed.git@main git+https://github.com/rapidsai/[email protected]"
+      test-before-amd64: "pip install git+https://github.com/dask/dask.git@2022.11.1 git+https://github.com/dask/distributed.git@2022.11.1 git+https://github.com/rapidsai/[email protected]"
       # On arm also need to install cupy from the specific webpage and CMake
       # because treelite needs to be compiled (no wheels available for arm).
-      test-before-arm64: "pip install cupy-cuda11x -f https://pip.cupy.dev/aarch64 && pip install cmake && pip install git+https://github.com/dask/dask.git@main git+https://github.com/dask/distributed.git@main git+https://github.com/rapidsai/[email protected]"
+      test-before-arm64: "pip install cupy-cuda11x -f https://pip.cupy.dev/aarch64 && pip install cmake && pip install git+https://github.com/dask/dask.git@2022.11.1 git+https://github.com/dask/distributed.git@2022.11.1 git+https://github.com/rapidsai/[email protected]"
       test-extras: test
       test-unittest: "pytest -v ./python/cuml/tests -k 'not test_silhouette_score_batched'"
     secrets: inherit
diff --git a/ci/gpu/build.sh b/ci/gpu/build.sh
@@ -39,10 +39,10 @@ export UCX_PY_VERSION='0.29.*'
 export NUMBA_THREADING_LAYER=workqueue
 
 # Whether to install dask nightly or stable packages
-export INSTALL_DASK_MAIN=1
+export INSTALL_DASK_MAIN=0
 
 # Dask version to install when `INSTALL_DASK_MAIN=0`
-export DASK_STABLE_VERSION="2022.9.2"
+export DASK_STABLE_VERSION="2022.11.1"
 
 ################################################################################
 # SETUP - Check environment
@@ -107,7 +107,7 @@ function install_dask {
         conda list
     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
+        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
     set +x
 }

diff --git a/conda/environments/cuml_dev_cuda11.0.yml b/conda/environments/cuml_dev_cuda11.0.yml
@@ -22,8 +22,8 @@ dependencies:
 - pylibraft=22.12.*
 - dask-cudf=22.12.*
 - dask-cuda=22.12.*
-- dask>=2022.9.2
-- distributed>=2022.9.2
+- dask==2022.11.1
+- distributed==2022.11.1
 - ucx>=1.13.0
 - ucx-py=0.29.*
 - ucx-proc=*=gpu

diff --git a/conda/environments/cuml_dev_cuda11.2.yml b/conda/environments/cuml_dev_cuda11.2.yml
@@ -22,8 +22,8 @@ dependencies:
 - pylibraft=22.12.*
 - dask-cudf=22.12.*
 - dask-cuda=22.12.*
-- dask>=2022.9.2
-- distributed>=2022.9.2
+- dask==2022.11.1
+- distributed==2022.11.1
 - ucx>=1.13.0
 - ucx-py=0.29.*
 - ucx-proc=*=gpu

diff --git a/conda/environments/cuml_dev_cuda11.4.yml b/conda/environments/cuml_dev_cuda11.4.yml
@@ -22,8 +22,8 @@ dependencies:
 - pylibraft=22.12.*
 - dask-cudf=22.12.*
 - dask-cuda=22.12.*
-- dask>=2022.9.2
-- distributed>=2022.9.2
+- dask==2022.11.1
+- distributed==2022.11.1
 - ucx>=1.13.0
 - ucx-py=0.29.*
 - ucx-proc=*=gpu

diff --git a/conda/environments/cuml_dev_cuda11.5.yml b/conda/environments/cuml_dev_cuda11.5.yml
@@ -22,8 +22,8 @@ dependencies:
 - pylibraft=22.12.*
 - dask-cudf=22.12.*
 - dask-cuda=22.12.*
-- dask>=2022.9.2
-- distributed>=2022.9.2
+- dask==2022.11.1
+- distributed==2022.11.1
 - ucx>=1.13.0
 - ucx-py=0.29.*
 - ucx-proc=*=gpu

diff --git a/conda/recipes/cuml/meta.yaml b/conda/recipes/cuml/meta.yaml
@@ -59,8 +59,8 @@ requirements:
     - nccl>=2.9.9
     - ucx-py {{ ucx_py_version }}
     - ucx-proc=*=gpu
-    - dask>=2022.9.2
-    - distributed>=2022.9.2
+    - dask==2022.11.1
+    - distributed==2022.11.1
     - joblib >=0.11
     - {{ pin_compatible('cudatoolkit', max_pin='x', min_pin='x') }}
     - cuda-python >=11.7.1,<12.0