Fix

hcho3 · Jun 23, 2024 · 084bff6 · 084bff6
1 parent 0adc883
commit 084bff6
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 6 deletions.
diff --git a/tests/buildkite/build-manylinux2014-aarch64.sh b/tests/buildkite/build-manylinux2014-aarch64.sh
@@ -6,21 +6,24 @@ source tests/buildkite/conftest.sh
 
 WHEEL_TAG=manylinux2014_aarch64
 command_wrapper="tests/ci_build/ci_build.sh manylinux2014_aarch64"
+python_bin="/opt/python/cp310-cp310/bin/python"
 
 echo "--- Build binary wheel for ${WHEEL_TAG}"
 $command_wrapper bash -c \
-  "cd python-package && pip wheel --no-deps -vvv . --wheel-dir dist/"
+  "cd python-package && ${python_bin} -m pip wheel --no-deps -vvv . --wheel-dir dist/"
 $command_wrapper auditwheel repair --plat ${WHEEL_TAG} python-package/dist/*.whl
-$command_wrapper python tests/ci_build/rename_whl.py  \
+$command_wrapper ${python_bin} tests/ci_build/rename_whl.py  \
   --wheel-path wheelhouse/*.whl  \
   --commit-hash ${BUILDKITE_COMMIT}  \
   --platform-tag ${WHEEL_TAG}
 mv -v wheelhouse/*.whl python-package/dist/
 
 echo "--- Upload Python wheel"
 buildkite-agent artifact upload python-package/dist/*.whl
+$command_wrapper bash -c "${python_bin} -m pip install awscli"
 if [[ ($is_pull_request == 0) && ($is_release_branch == 1) ]]
 then
-  aws s3 cp python-package/dist/*.whl s3://xgboost-nightly-builds/${BRANCH_NAME}/ \
+  ${python_bin} -m awscli s3 \
+    cp python-package/dist/*.whl s3://xgboost-nightly-builds/${BRANCH_NAME}/ \
     --acl public-read --no-progress
 fi
diff --git a/tests/buildkite/build-manylinux2014-x86_64.sh b/tests/buildkite/build-manylinux2014-x86_64.sh
@@ -6,21 +6,24 @@ source tests/buildkite/conftest.sh
 
 WHEEL_TAG=manylinux2014_x86_64
 command_wrapper="tests/ci_build/ci_build.sh manylinux2014_x86_64"
+python_bin="/opt/python/cp310-cp310/bin/python"
 
 echo "--- Build binary wheel for ${WHEEL_TAG}"
 $command_wrapper bash -c \
-  "cd python-package && pip wheel --no-deps -vvv . --wheel-dir dist/"
+  "cd python-package && ${python_bin} -m pip wheel --no-deps -vvv . --wheel-dir dist/"
 $command_wrapper auditwheel repair --plat ${WHEEL_TAG} python-package/dist/*.whl
-$command_wrapper python tests/ci_build/rename_whl.py  \
+$command_wrapper ${python_bin} tests/ci_build/rename_whl.py  \
   --wheel-path wheelhouse/*.whl  \
   --commit-hash ${BUILDKITE_COMMIT}  \
   --platform-tag ${WHEEL_TAG}
 mv -v wheelhouse/*.whl python-package/dist/
 
 echo "--- Upload Python wheel"
 buildkite-agent artifact upload python-package/dist/*.whl
+$command_wrapper bash -c "${python_bin} -m pip install awscli"
 if [[ ($is_pull_request == 0) && ($is_release_branch == 1) ]]
 then
-  aws s3 cp python-package/dist/*.whl s3://xgboost-nightly-builds/${BRANCH_NAME}/ \
+  ${python_bin} -m awscli s3 \
+    cp python-package/dist/*.whl s3://xgboost-nightly-builds/${BRANCH_NAME}/ \
     --acl public-read --no-progress
 fi