From 10f3e1734781a2c0878d25ed377d8355303c7988 Mon Sep 17 00:00:00 2001 From: Gil Forsyth Date: Wed, 5 Feb 2025 10:09:28 -0500 Subject: [PATCH] Use `rapids-pip-retry` in CI jobs that might need retries --- ci/build_wheel.sh | 2 +- ci/test_wheel.sh | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/ci/build_wheel.sh b/ci/build_wheel.sh index 741aeab90..c43718437 100755 --- a/ci/build_wheel.sh +++ b/ci/build_wheel.sh @@ -7,7 +7,7 @@ source rapids-date-string rapids-generate-version > ./VERSION -python -m pip wheel . -w dist -v --no-deps --disable-pip-version-check +rapids-pip-retry wheel . -w dist -v --no-deps --disable-pip-version-check ./ci/validate_wheel.sh dist RAPIDS_PY_WHEEL_NAME="dask-cuda" RAPIDS_PY_WHEEL_PURE="1" rapids-upload-wheels-to-s3 dist diff --git a/ci/test_wheel.sh b/ci/test_wheel.sh index 9d5948a3d..4214dd2e0 100755 --- a/ci/test_wheel.sh +++ b/ci/test_wheel.sh @@ -14,7 +14,7 @@ rapids-dependency-file-generator \ rapids-logger "Installing test dependencies" # echo to expand wildcard -python -m pip install -v --prefer-binary -r /tmp/requirements-test.txt "$(echo ./dist/dask_cuda*.whl)" +rapids-pip-retry install -v --prefer-binary -r /tmp/requirements-test.txt "$(echo ./dist/dask_cuda*.whl)" rapids-logger "pytest dask-cuda" ./ci/run_pytest.sh \