From 5676fd5a3b2548980649a3880185e3cf073e99d2 Mon Sep 17 00:00:00 2001 From: GALI PREM SAGAR Date: Wed, 23 Mar 2022 08:10:26 -0500 Subject: [PATCH] Pin `dask` & `distributed` versions (#4647) Pin changes to be in-line with: https://github.com/rapidsai/cudf/pull/10481/ Authors: - GALI PREM SAGAR (https://github.com/galipremsagar) Approvers: - AJ Schmidt (https://github.com/ajschmidt8) - Dante Gama Dessavre (https://github.com/dantegd) URL: https://github.com/rapidsai/cuml/pull/4647 --- ci/gpu/build.sh | 8 ++++---- conda/environments/cuml_dev_cuda11.0.yml | 4 ++-- conda/environments/cuml_dev_cuda11.2.yml | 4 ++-- conda/environments/cuml_dev_cuda11.4.yml | 4 ++-- conda/environments/cuml_dev_cuda11.5.yml | 4 ++-- conda/recipes/cuml/meta.yaml | 4 ++-- 6 files changed, 14 insertions(+), 14 deletions(-) diff --git a/ci/gpu/build.sh b/ci/gpu/build.sh index c7369f7643..4bfc154c8b 100755 --- a/ci/gpu/build.sh +++ b/ci/gpu/build.sh @@ -127,8 +127,8 @@ if [[ -z "$PROJECT_FLASH" || "$PROJECT_FLASH" == "0" ]]; then gpuci_logger "Install the main version of dask and distributed" set -x - pip install "git+https://github.com/dask/distributed.git@main" --upgrade --no-deps - pip install "git+https://github.com/dask/dask.git@main" --upgrade --no-deps + pip install "git+https://github.com/dask/distributed.git@2022.03.0" --upgrade --no-deps + pip install "git+https://github.com/dask/dask.git@2022.03.0" --upgrade --no-deps set +x gpuci_logger "Python pytest for cuml" @@ -222,8 +222,8 @@ else gpuci_logger "Install the main version of dask and distributed" set -x - pip install "git+https://github.com/dask/distributed.git@main" --upgrade --no-deps - pip install "git+https://github.com/dask/dask.git@main" --upgrade --no-deps + pip install "git+https://github.com/dask/distributed.git@2022.03.0" --upgrade --no-deps + pip install "git+https://github.com/dask/dask.git@2022.03.0" --upgrade --no-deps set +x gpuci_logger "Python pytest for cuml" diff --git a/conda/environments/cuml_dev_cuda11.0.yml b/conda/environments/cuml_dev_cuda11.0.yml index f6b35a48da..f0221b8956 100644 --- a/conda/environments/cuml_dev_cuda11.0.yml +++ b/conda/environments/cuml_dev_cuda11.0.yml @@ -35,8 +35,8 @@ dependencies: - pip - pip: - sphinx_markdown_tables - - git+https://github.com/dask/dask.git@main - - git+https://github.com/dask/distributed.git@main + - git+https://github.com/dask/dask.git@2022.03.0 + - git+https://github.com/dask/distributed.git@2022.03.0 # rapids-build-env, notebook-env and doc-env are defined in # https://docs.rapids.ai/maintainers/depmgmt/ diff --git a/conda/environments/cuml_dev_cuda11.2.yml b/conda/environments/cuml_dev_cuda11.2.yml index 5ff50599ec..eaf47bf649 100644 --- a/conda/environments/cuml_dev_cuda11.2.yml +++ b/conda/environments/cuml_dev_cuda11.2.yml @@ -35,8 +35,8 @@ dependencies: - pip - pip: - sphinx_markdown_tables - - git+https://github.com/dask/dask.git@main - - git+https://github.com/dask/distributed.git@main + - git+https://github.com/dask/dask.git@2022.03.0 + - git+https://github.com/dask/distributed.git@2022.03.0 # rapids-build-env, notebook-env and doc-env are defined in # https://docs.rapids.ai/maintainers/depmgmt/ diff --git a/conda/environments/cuml_dev_cuda11.4.yml b/conda/environments/cuml_dev_cuda11.4.yml index 610a3fb649..9b83a1e038 100644 --- a/conda/environments/cuml_dev_cuda11.4.yml +++ b/conda/environments/cuml_dev_cuda11.4.yml @@ -35,8 +35,8 @@ dependencies: - pip - pip: - sphinx_markdown_tables - - git+https://github.com/dask/dask.git@main - - git+https://github.com/dask/distributed.git@main + - git+https://github.com/dask/dask.git@2022.03.0 + - git+https://github.com/dask/distributed.git@2022.03.0 # rapids-build-env, notebook-env and doc-env are defined in # https://docs.rapids.ai/maintainers/depmgmt/ diff --git a/conda/environments/cuml_dev_cuda11.5.yml b/conda/environments/cuml_dev_cuda11.5.yml index 210c3af2fd..c986c10487 100644 --- a/conda/environments/cuml_dev_cuda11.5.yml +++ b/conda/environments/cuml_dev_cuda11.5.yml @@ -35,8 +35,8 @@ dependencies: - pip - pip: - sphinx_markdown_tables - - git+https://github.com/dask/dask.git@main - - git+https://github.com/dask/distributed.git@main + - git+https://github.com/dask/dask.git@2022.03.0 + - git+https://github.com/dask/distributed.git@2022.03.0 # rapids-build-env, notebook-env and doc-env are defined in # https://docs.rapids.ai/maintainers/depmgmt/ diff --git a/conda/recipes/cuml/meta.yaml b/conda/recipes/cuml/meta.yaml index 0d99edeb5a..67a476732b 100644 --- a/conda/recipes/cuml/meta.yaml +++ b/conda/recipes/cuml/meta.yaml @@ -51,8 +51,8 @@ requirements: - nccl>=2.9.9 - ucx-py {{ ucx_py_version }} - ucx-proc=*=gpu - - dask>=2022.02.1 - - distributed>=2022.02.1 + - dask==2022.03.0 + - distributed==2022.03.0 - joblib >=0.11 - {{ pin_compatible('cudatoolkit', max_pin='x', min_pin='x') }} - cuda-python >=11.5,<12.0