rapids-get-pr-wheel-artifact was missing RAPIDS_PY_WHEEL_NAME

rapidsai · Sep 17, 2024 · d4ef11a · d4ef11a
1 parent f1b0793
commit d4ef11a
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 6 deletions.
diff --git a/ci/build_wheel.sh b/ci/build_wheel.sh
@@ -39,8 +39,12 @@ case "${RAPIDS_CUDA_VERSION}" in
   ;;
 esac
 
-LIBRMM_CHANNEL=$(rapids-get-artifact "ci/rmm/pull-request/1678/4127d9c/rmm_wheel_cpp_rmm_${RAPIDS_PY_CUDA_SUFFIX}_x86_64.tar.gz")
-RMM_CHANNEL=$(rapids-get-artifact "ci/rmm/pull-request/1678/4127d9c/rmm_wheel_python_rmm_${RAPIDS_PY_CUDA_SUFFIX}_py${RAPIDS_PY_VERSION//./}_x86_64.tar.gz")
+LIBRMM_CHANNEL=$(
+  RAPIDS_PY_WHEEL_NAME=rmm_${RAPIDS_PY_CUDA_SUFFIX} rapids-get-pr-wheel-artifact rmm 1678 cpp
+)
+RMM_CHANNEL=$(
+  RAPIDS_PY_WHEEL_NAME=rmm_${RAPIDS_PY_CUDA_SUFFIX}_py${RAPIDS_PY_VERSION//./} rapids-get-pr-wheel-artifact rmm 1678 python
+)
 
 echo "rmm-${RAPIDS_PY_CUDA_SUFFIX} @ $(echo ${RMM_CHANNEL}/rmm*.whl)" > /tmp/constraints.txt
 echo "librmm-${RAPIDS_PY_CUDA_SUFFIX} @ $(echo ${LIBRMM_CHANNEL}/librmm*.whl)" >> /tmp/constraints.txt

diff --git a/ci/test_wheel_pylibraft.sh b/ci/test_wheel_pylibraft.sh
@@ -7,8 +7,12 @@ mkdir -p ./dist
 RAPIDS_PY_CUDA_SUFFIX="$(rapids-wheel-ctk-name-gen ${RAPIDS_CUDA_VERSION})"
 RAPIDS_PY_WHEEL_NAME="pylibraft_${RAPIDS_PY_CUDA_SUFFIX}" rapids-download-wheels-from-s3 ./dist
 
-LIBRMM_CHANNEL=$(rapids-get-artifact "ci/rmm/pull-request/1678/4127d9c/rmm_wheel_cpp_rmm_${RAPIDS_PY_CUDA_SUFFIX}_x86_64.tar.gz")
-RMM_CHANNEL=$(rapids-get-artifact "ci/rmm/pull-request/1678/4127d9c/rmm_wheel_python_rmm_${RAPIDS_PY_CUDA_SUFFIX}_py${RAPIDS_PY_VERSION//./}_x86_64.tar.gz")
+LIBRMM_CHANNEL=$(
+  RAPIDS_PY_WHEEL_NAME=rmm_${RAPIDS_PY_CUDA_SUFFIX} rapids-get-pr-wheel-artifact rmm 1678 cpp
+)
+RMM_CHANNEL=$(
+  RAPIDS_PY_WHEEL_NAME=rmm_${RAPIDS_PY_CUDA_SUFFIX}_py${RAPIDS_PY_VERSION//./} rapids-get-pr-wheel-artifact rmm 1678 python
+)
 
 echo "rmm-${RAPIDS_PY_CUDA_SUFFIX} @ $(echo ${RMM_CHANNEL}/rmm*.whl)" > /tmp/constraints.txt
 echo "librmm-${RAPIDS_PY_CUDA_SUFFIX} @ $(echo ${LIBRMM_CHANNEL}/librmm*.whl)" >> /tmp/constraints.txt

diff --git a/ci/test_wheel_raft_dask.sh b/ci/test_wheel_raft_dask.sh
@@ -7,8 +7,12 @@ mkdir -p ./dist
 RAPIDS_PY_CUDA_SUFFIX="$(rapids-wheel-ctk-name-gen ${RAPIDS_CUDA_VERSION})"
 RAPIDS_PY_WHEEL_NAME="raft_dask_${RAPIDS_PY_CUDA_SUFFIX}" rapids-download-wheels-from-s3 ./dist
 
-LIBRMM_CHANNEL=$(rapids-get-artifact "ci/rmm/pull-request/1678/4127d9c/rmm_wheel_cpp_rmm_${RAPIDS_PY_CUDA_SUFFIX}_x86_64.tar.gz")
-RMM_CHANNEL=$(rapids-get-artifact "ci/rmm/pull-request/1678/4127d9c/rmm_wheel_python_rmm_${RAPIDS_PY_CUDA_SUFFIX}_py${RAPIDS_PY_VERSION//./}_x86_64.tar.gz")
+LIBRMM_CHANNEL=$(
+  RAPIDS_PY_WHEEL_NAME=rmm_${RAPIDS_PY_CUDA_SUFFIX} rapids-get-pr-wheel-artifact rmm 1678 cpp
+)
+RMM_CHANNEL=$(
+  RAPIDS_PY_WHEEL_NAME=rmm_${RAPIDS_PY_CUDA_SUFFIX}_py${RAPIDS_PY_VERSION//./} rapids-get-pr-wheel-artifact rmm 1678 python
+)
 
 echo "rmm-${RAPIDS_PY_CUDA_SUFFIX} @ $(echo ${RMM_CHANNEL}/rmm*.whl)" > /tmp/constraints.txt
 echo "librmm-${RAPIDS_PY_CUDA_SUFFIX} @ $(echo ${LIBRMM_CHANNEL}/librmm*.whl)" >> /tmp/constraints.txt