Skip to content

Commit e18c575

Browse files
authored
Use extra-index-url for linux-aarch64 testing (#4536)
These wheels are published on pypi officially hence no reason to test with index-url <!-- copilot:poem --> ### <samp>🤖 Generated by Copilot at 31a9def</samp> > _We're sailing on the Python sea, with wheels of every kind_ > _We need to use the right pip option, or else we'll fall behind_ > _So `get_wheel_install_command` will check the OS for us_ > _And set the variable `option` with a minimum of fuss_
1 parent b7ba526 commit e18c575

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tools/scripts/generate_binary_build_matrix.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -256,6 +256,8 @@ def get_wheel_install_command(
256256
desired_cuda: str,
257257
python_version: str,
258258
) -> str:
259+
260+
index_url_option = "--index-url" if os != "linux-aarch64" else "--extra-index-url"
259261
if channel == RELEASE and (
260262
(gpu_arch_version == "11.7" and os == "linux")
261263
or (
@@ -270,7 +272,7 @@ def get_wheel_install_command(
270272
if channel == "nightly"
271273
else f"{WHL_INSTALL_BASE} {PACKAGES_TO_INSTALL_WHL}"
272274
)
273-
return f"{whl_install_command} --index-url {get_base_download_url_for_repo('whl', channel, gpu_arch_type, desired_cuda)}"
275+
return f"{whl_install_command} {index_url_option} {get_base_download_url_for_repo('whl', channel, gpu_arch_type, desired_cuda)}"
274276

275277

276278
def generate_conda_matrix(

0 commit comments

Comments
 (0)