diff --git a/ci/gpu/build.sh b/ci/gpu/build.sh index 9d8feebb14..4fee6ebcf4 100755 --- a/ci/gpu/build.sh +++ b/ci/gpu/build.sh @@ -122,8 +122,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@2021.05.1" --upgrade --no-deps - pip install "git+https://github.com/dask/dask.git@2021.05.1" --upgrade --no-deps + 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 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 5090d1aabe..55157accbc 100644 --- a/conda/environments/cuml_dev_cuda11.0.yml +++ b/conda/environments/cuml_dev_cuda11.0.yml @@ -26,8 +26,8 @@ dependencies: - pip - pip: - sphinx_markdown_tables - - git+https://github.com/dask/dask.git@2021.05.1 - - git+https://github.com/dask/distributed.git@2021.05.1 + - git+https://github.com/dask/dask.git@main + - git+https://github.com/dask/distributed.git@main # 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 bc75ae9967..1d41be917e 100644 --- a/conda/environments/cuml_dev_cuda11.2.yml +++ b/conda/environments/cuml_dev_cuda11.2.yml @@ -26,8 +26,8 @@ dependencies: - pip - pip: - sphinx_markdown_tables - - git+https://github.com/dask/dask.git@2021.05.1 - - git+https://github.com/dask/distributed.git@2021.05.1 + - git+https://github.com/dask/dask.git@main + - git+https://github.com/dask/distributed.git@main # rapids-build-env, notebook-env and doc-env are defined in # https://docs.rapids.ai/maintainers/depmgmt/