Skip to content

Commit

Permalink
add pkg_type to rapids-download_wheels_from_s3
Browse files Browse the repository at this point in the history
  • Loading branch information
msarahan committed Apr 2, 2024
1 parent 2b50d2a commit 076d082
Showing 1 changed file with 10 additions and 1 deletion.
11 changes: 10 additions & 1 deletion tools/rapids-download-wheels-from-s3
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,15 @@ set -eo pipefail

source rapids-prompt-local-repo-config

pkg_name="$(rapids-package-name wheel_python)"
pkg_type="$1"
if [ "${pkg_type}" != "cpp" ] && [ "${pkg_type}" != "python" ]; then
rapids-echo-error "Wheel cpp packages are now supported."
rapids-echo-error "You must specify package type as the first argument to rapids-upload-wheels-to-s3"
fi

# remove pkg_type from args because we handle it in this script
shift;

pkg_name="$(rapids-package-name wheel_${pkg_type})"

rapids-download-from-s3 "${pkg_name}" "$@"

0 comments on commit 076d082

Please sign in to comment.