From 940846ad4ff7106ba46f924492eb62552c5873f0 Mon Sep 17 00:00:00 2001
From: Kyle Edwards <kyedwards@nvidia.com>
Date: Fri, 14 Feb 2025 17:29:14 -0500
Subject: [PATCH] Consolidate more Conda solves in CI (#1452)

Issue: https://github.com/rapidsai/build-planning/issues/22

Authors:
  - Kyle Edwards (https://github.com/KyleFromNVIDIA)

Approvers:
  - James Lamb (https://github.com/jameslamb)

URL: https://github.com/rapidsai/dask-cuda/pull/1452
---
 ci/build_docs.sh  | 18 +++++++-----------
 dependencies.yaml |  1 +
 2 files changed, 8 insertions(+), 11 deletions(-)

diff --git a/ci/build_docs.sh b/ci/build_docs.sh
index 5efa064ce..1034446eb 100755
--- a/ci/build_docs.sh
+++ b/ci/build_docs.sh
@@ -1,30 +1,26 @@
 #!/bin/bash
+# Copyright (c) 2023-2025, NVIDIA CORPORATION.
 
 set -euo pipefail
 
+rapids-logger "Downloading artifacts from previous jobs"
+PYTHON_CHANNEL=$(rapids-download-conda-from-s3 python)
+
 rapids-logger "Create test conda environment"
 . /opt/conda/etc/profile.d/conda.sh
 
-RAPIDS_VERSION="$(rapids-version)"
-
 rapids-dependency-file-generator \
     --output conda \
     --file-key docs \
-    --matrix "cuda=${RAPIDS_CUDA_VERSION%.*};arch=$(arch);py=${RAPIDS_PY_VERSION}" | tee env.yaml
+    --matrix "cuda=${RAPIDS_CUDA_VERSION%.*};arch=$(arch);py=${RAPIDS_PY_VERSION}" \
+    --prepend-channel "${PYTHON_CHANNEL}" \
+    | tee env.yaml
 
 rapids-mamba-retry env create --yes -f env.yaml -n docs
 conda activate docs
 
 rapids-print-env
 
-rapids-logger "Downloading artifacts from previous jobs"
-
-PYTHON_CHANNEL=$(rapids-download-conda-from-s3 python)
-
-rapids-mamba-retry install \
-    --channel "${PYTHON_CHANNEL}" \
-    "dask-cuda=${RAPIDS_VERSION}"
-
 RAPIDS_DOCS_DIR="$(mktemp -d)"
 export RAPIDS_DOCS_DIR
 
diff --git a/dependencies.yaml b/dependencies.yaml
index f5954e36e..7eda41d49 100644
--- a/dependencies.yaml
+++ b/dependencies.yaml
@@ -34,6 +34,7 @@ files:
       - cuda_version
       - docs
       - py_version
+      - depends_on_dask_cuda
   py_build:
     output: pyproject
     pyproject_dir: .