From f6cd7100249d71fa0991ac83c9616237df7ef425 Mon Sep 17 00:00:00 2001 From: James Lamb Date: Thu, 1 Oct 2026 10:42:57 -0500 Subject: [PATCH] pip: use proxy cache for 'rapidsai-wheels-nightly' in CI --- tools/rapids-init-pip | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/tools/rapids-init-pip b/tools/rapids-init-pip index a88c506..90144e1 100755 --- a/tools/rapids-init-pip +++ b/tools/rapids-init-pip @@ -28,3 +28,9 @@ export PIP_CONSTRAINT PIP_CACHE_DIR="${PIP_CACHE_DIR:-${GITHUB_WORKSPACE:-${PWD}}/.cache/pip}" mkdir -p "${PIP_CACHE_DIR}" export PIP_CACHE_DIR + +# activate NVIDIA-hosted proxy cache(s) +if [[ -n "${NV_GHA_RUNNERS_PYPI_ANACONDA_CACHE_URL:-}" ]]; then + # NOTE: this intentionally overwrites 'extra-index-url' in 'pip.conf' that comes installed in CI images + pip config --global set global.extra-index-url "${NV_GHA_RUNNERS_PYPI_ANACONDA_CACHE_URL}/rapidsai-wheels-nightly/simple" +fi