From 94ca788e9441f24fe4c70801dbf87fa8c6c264dc Mon Sep 17 00:00:00 2001
From: James Lamb <jlamb@nvidia.com>
Date: Wed, 4 Mar 2026 14:57:59 -0600
Subject: [PATCH 1/2] wheels CI: write constraints directly to PIP_CONSTRAINT

---
 ci/test_wheel.sh      | 10 ++++------
 ci/test_wheel_dask.sh |  9 ++++-----
 2 files changed, 8 insertions(+), 11 deletions(-)

diff --git a/ci/test_wheel.sh b/ci/test_wheel.sh
index 75a07c5afd..84cb65d128 100755
--- a/ci/test_wheel.sh
+++ b/ci/test_wheel.sh
@@ -14,13 +14,12 @@ mkdir -p "${RAPIDS_TESTS_DIR}"
 
 # generate constraints, the constraints will limit the version of the
 # dependencies that can be installed later on when installing the wheel
-rapids-generate-pip-constraints test_python ./constraints.txt
+rapids-generate-pip-constraints test_python "${PIP_CONSTRAINT}"
 
 # Install just minimal dependencies first
 rapids-pip-retry install \
   "${LIBCUML_WHEELHOUSE}"/libcuml*.whl \
   "${CUML_WHEELHOUSE}"/cuml*.whl \
-  --constraint ./constraints.txt \
   --constraint "${PIP_CONSTRAINT}"
 
 # Try to import cuml with just a minimal install"
@@ -29,14 +28,13 @@ python -c "import cuml"
 
 # notes:
 #
-#   * echo to expand wildcard before adding `[test,experimental]` requires for pip
-#   * need to provide --constraint="${PIP_CONSTRAINT}" because that environment variable is
-#     ignored if any other --constraint are passed via the CLI
+#   * echo to expand wildcard before adding `[test]` requires for pip
+#   * just providing --constraint="${PIP_CONSTRAINT}" to be explicit, and because
+#     that environment variable is ignored if any other --constraint are passed via the CLI
 #
 rapids-pip-retry install \
    "${LIBCUML_WHEELHOUSE}"/libcuml*.whl \
   "$(echo "${CUML_WHEELHOUSE}"/cuml*.whl)[test]" \
-  --constraint ./constraints.txt \
   --constraint "${PIP_CONSTRAINT}"
 
 EXITCODE=0
diff --git a/ci/test_wheel_dask.sh b/ci/test_wheel_dask.sh
index bbdded3d08..bd336c9833 100755
--- a/ci/test_wheel_dask.sh
+++ b/ci/test_wheel_dask.sh
@@ -14,18 +14,17 @@ mkdir -p "${RAPIDS_TESTS_DIR}"
 
 # generate constraints, the constraints will limit the version of the
 # dependencies that can be installed later on when installing the wheel
-rapids-generate-pip-constraints test_python ./constraints.txt
+rapids-generate-pip-constraints test_python "${PIP_CONSTRAINT}"
 
 # notes:
 #
-#   * echo to expand wildcard before adding `[...]` extras for pip
-#   * need to provide --constraint="${PIP_CONSTRAINT}" because that environment variable is
-#     ignored if any other --constraint are passed via the CLI
+#   * echo to expand wildcard before adding `[...]` requires for pip
+#   * just providing --constraint="${PIP_CONSTRAINT}" to be explicit, and because
+#     that environment variable is ignored if any other --constraint are passed via the CLI
 #
 rapids-pip-retry install \
    "${LIBCUML_WHEELHOUSE}"/libcuml*.whl \
   "$(echo "${CUML_WHEELHOUSE}"/cuml*.whl)[dask,test,test-dask]" \
-  --constraint ./constraints.txt \
   --constraint "${PIP_CONSTRAINT}"
 
 EXITCODE=0

From b541294e498375ecf3002c072feea2f6470a0a43 Mon Sep 17 00:00:00 2001
From: James Lamb <jlamb@nvidia.com>
Date: Wed, 4 Mar 2026 15:01:14 -0600
Subject: [PATCH 2/2] use fork

---
 ci/test_wheel.sh      | 7 +++++++
 ci/test_wheel_dask.sh | 7 +++++++
 2 files changed, 14 insertions(+)

diff --git a/ci/test_wheel.sh b/ci/test_wheel.sh
index 84cb65d128..483fcd3e23 100755
--- a/ci/test_wheel.sh
+++ b/ci/test_wheel.sh
@@ -4,6 +4,13 @@
 
 set -euo pipefail
 
+# TODO(jameslamb): revert before merging
+git clone --branch generate-pip-constraints \
+    https://github.com/rapidsai/gha-tools.git \
+    /tmp/gha-tools
+
+export PATH="/tmp/gha-tools/tools:${PATH}"
+
 source rapids-init-pip
 
 RAPIDS_PY_CUDA_SUFFIX="$(rapids-wheel-ctk-name-gen "${RAPIDS_CUDA_VERSION}")"
diff --git a/ci/test_wheel_dask.sh b/ci/test_wheel_dask.sh
index bd336c9833..4bedb45724 100755
--- a/ci/test_wheel_dask.sh
+++ b/ci/test_wheel_dask.sh
@@ -4,6 +4,13 @@
 
 set -euo pipefail
 
+# TODO(jameslamb): revert before merging
+git clone --branch generate-pip-constraints \
+    https://github.com/rapidsai/gha-tools.git \
+    /tmp/gha-tools
+
+export PATH="/tmp/gha-tools/tools:${PATH}"
+
 source rapids-init-pip
 
 RAPIDS_PY_CUDA_SUFFIX="$(rapids-wheel-ctk-name-gen "${RAPIDS_CUDA_VERSION}")"