From a38cdc2d06b515143f4cea56b844d5b571ecf65f Mon Sep 17 00:00:00 2001
From: Gil Forsyth <gforsyth@nvidia.com>
Date: Wed, 5 Feb 2025 09:54:57 -0500
Subject: [PATCH] Use `rapids-pip-retry` in CI jobs that might need retries

---
 ci/build_wheel.sh | 4 ++--
 ci/test_wheel.sh  | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/ci/build_wheel.sh b/ci/build_wheel.sh
index 112cf0a4..da560074 100755
--- a/ci/build_wheel.sh
+++ b/ci/build_wheel.sh
@@ -25,7 +25,7 @@ rapids-dependency-file-generator \
 | tee /tmp/requirements-build.txt
 
 rapids-logger "Installing build requirements"
-python -m pip install \
+rapids-pip-retry install \
     -v \
     --prefer-binary \
     -r /tmp/requirements-build.txt
@@ -42,7 +42,7 @@ cd "${package_dir}"
 sccache --zero-stats
 
 rapids-logger "Building '${package_name}' wheel"
-python -m pip wheel \
+rapids-pip-retry wheel \
     -w dist \
     -v \
     --no-build-isolation \
diff --git a/ci/test_wheel.sh b/ci/test_wheel.sh
index af0f6bcc..4d9c69af 100755
--- a/ci/test_wheel.sh
+++ b/ci/test_wheel.sh
@@ -7,7 +7,7 @@ RAPIDS_PY_CUDA_SUFFIX="$(rapids-wheel-ctk-name-gen "${RAPIDS_CUDA_VERSION}")"
 RAPIDS_PY_WHEEL_NAME="cucim_${RAPIDS_PY_CUDA_SUFFIX}" rapids-download-wheels-from-s3 ./dist
 
 # echo to expand wildcard before adding `[extra]` requires for pip
-python -m pip install "$(echo ./dist/cucim*.whl)[test]"
+rapids-pip-retry install "$(echo ./dist/cucim*.whl)[test]"
 
 CUDA_MAJOR_VERSION=${RAPIDS_CUDA_VERSION:0:2}