From 159b5f72b4b04e11642845804f6de8fbd01397a2 Mon Sep 17 00:00:00 2001
From: jakirkham <jakirkham@gmail.com>
Date: Tue, 7 Jan 2025 20:46:20 -0800
Subject: [PATCH] Use `rapids-dask-dependency`

Switch to use `rapids-dask-dependency` to handle the Dask test
dependency in this repo. That way it aligns to how Dask is pulled and
used in the rest of RAPIDS and will pick up any corresponding updates
simply.
---
 ci/release/update-version.sh | 1 +
 dependencies.yaml            | 2 +-
 pyproject.toml               | 2 +-
 3 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/ci/release/update-version.sh b/ci/release/update-version.sh
index 559d235d..1863bfc5 100755
--- a/ci/release/update-version.sh
+++ b/ci/release/update-version.sh
@@ -40,6 +40,7 @@ function sed_runner() {
 
 DEPENDENCIES=(
   cudf
+  rapids-dask-dependency
 )
 for DEP in "${DEPENDENCIES[@]}"; do
   for FILE in dependencies.yaml conda/environments/*.yml; do
diff --git a/dependencies.yaml b/dependencies.yaml
index f9e59df4..107332d1 100644
--- a/dependencies.yaml
+++ b/dependencies.yaml
@@ -217,7 +217,7 @@ dependencies:
       - output_types: [conda, requirements, pyproject]
         packages:
           - cloudpickle
-          - dask
+          - rapids-dask-dependency==25.2.*,>=0.0.0a0
           - distributed
           - numba>=0.57
           - pytest==7.*
diff --git a/pyproject.toml b/pyproject.toml
index 94174b58..1ebe1d33 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -50,12 +50,12 @@ test = [
     "cloudpickle",
     "cudf==25.2.*,>=0.0.0a0",
     "cupy-cuda12x>=12.0.0",
-    "dask",
     "distributed",
     "numba>=0.57",
     "pytest-asyncio",
     "pytest-rerunfailures",
     "pytest==7.*",
+    "rapids-dask-dependency==25.2.*,>=0.0.0a0",
 ] # This list was generated by `rapids-dependency-file-generator`. To make changes, edit dependencies.yaml and run `rapids-dependency-file-generator`.
 
 [project.urls]