34 lines
930 B
Bash
Executable File
34 lines
930 B
Bash
Executable File
#!/usr/bin/env bash
|
|
|
|
set -ex
|
|
|
|
# Upload a single wheel to S3 under the omni/ prefix.
|
|
# Index generation is handled separately by generate-and-upload-nightly-index.sh.
|
|
|
|
BUCKET="vllm-wheels"
|
|
SUBPATH="omni/$BUILDKITE_COMMIT"
|
|
S3_COMMIT_PREFIX="s3://$BUCKET/$SUBPATH/"
|
|
|
|
# ========= collect & upload the wheel ==========
|
|
|
|
# python3 -m build outputs to dist/ by default
|
|
wheel_files=(dist/*.whl)
|
|
|
|
# Check that exactly one wheel is found
|
|
if [[ ${#wheel_files[@]} -ne 1 ]]; then
|
|
echo "Error: Expected exactly one wheel file in dist/, but found ${#wheel_files[@]}"
|
|
exit 1
|
|
fi
|
|
wheel="${wheel_files[0]}"
|
|
|
|
echo "Uploading wheel: $wheel"
|
|
|
|
# Extract the version from the wheel
|
|
version=$(unzip -p "$wheel" '**/METADATA' | grep '^Version: ' | cut -d' ' -f2)
|
|
echo "Version in wheel: $version"
|
|
|
|
# Upload wheel to S3
|
|
aws s3 cp "$wheel" "$S3_COMMIT_PREFIX"
|
|
|
|
echo "Wheel uploaded to $S3_COMMIT_PREFIX. Index generation is handled by a separate step."
|