From e0b9798e6fa1d79ce116f55d0f716797d89802d2 Mon Sep 17 00:00:00 2001
From: James Lamb <jlamb@nvidia.com>
Date: Thu, 23 Jan 2025 15:55:52 -0600
Subject: [PATCH] ensure update-version.sh updates pyproject.toml (#1106)

Contributes to https://github.com/rapidsai/build-planning/issues/136

This project's `update-version.sh` misses some RAPIDS dependencies in `pyproject.toml`. This fixes that.

Also updates to the latest `rapids-dependency-file-generator`.

## Notes for Reviewers

### How I tested this

```shell
git fetch upstream --tags
./ci/release/update-version.sh '0.43.00'
git grep -E '25\.2'
git grep -E '25\.02'
git grep -E '0\.42'
```

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

Approvers:
  - Jake Awe (https://github.com/AyodeAwe)

URL: https://github.com/rapidsai/ucx-py/pull/1106
---
 .pre-commit-config.yaml      | 2 +-
 ci/release/update-version.sh | 4 ++++
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
index 66e7f889..2c2b7f11 100644
--- a/.pre-commit-config.yaml
+++ b/.pre-commit-config.yaml
@@ -27,7 +27,7 @@ repos:
                   - --fix
                   - --rapids-version=25.02
       - repo: https://github.com/rapidsai/dependency-file-generator
-        rev: v1.16.0
+        rev: v1.17.0
         hooks:
               - id: rapids-dependency-file-generator
                 args: ["--clean"]
diff --git a/ci/release/update-version.sh b/ci/release/update-version.sh
index ed22e1a9..d023c034 100755
--- a/ci/release/update-version.sh
+++ b/ci/release/update-version.sh
@@ -54,6 +54,10 @@ for FILE in dependencies.yaml conda/environments/*.yml; do
   done
 done
 
+for DEP in "${DEPENDENCIES[@]}"; do
+  sed_runner "/\"${DEP}\(-cu[[:digit:]]\{2\}\)\{0,1\}==/ s/==.*\"/==${NEXT_RAPIDS_SHORT_TAG_PEP440}.*,>=0.0.0a0\"/g" pyproject.toml
+done
+
 for FILE in .github/workflows/*.yaml; do
   sed_runner "/shared-workflows/ s/@.*/@branch-${NEXT_RAPIDS_SHORT_TAG}/g" "${FILE}"
 done