A high-throughput and memory-efficient inference and serving engine for LLMs
Mirrored from https://github.com/tduncan-lum/Nvidia-Optimization-Stack---VLLM.git
Synced Nov 7, 05:15 UTC
[Release] Include source distribution (sdist) in PyPI uploads (#35136)
Signed-off-by: dougbtv <dosmith@redhat.com> Co-authored-by: Daniele Trifirò <dtrifiro@redhat.com>
Doug Smith committed
Mar 5, 2026 at 04:43 UTC
0bfa229bf1f6b12f215d045f4acb4b9607937f32
1 file changed
+8
-5
.buildkite/scripts/upload-release-wheels-pypi.sh
+8
-5
@@ -54,10 +54,13 @@ mkdir -p $DIST_DIR
54
# include only wheels for the release version, ignore all files with "dev" or "rc" in the name (without excluding 'aarch64')
55
aws s3 cp --recursive --exclude "*" --include "vllm-${PURE_VERSION}*.whl" --exclude "*dev*" --exclude "*rc[0-9]*" "$S3_COMMIT_PREFIX" $DIST_DIR
56
echo "Wheels copied to local directory"
57
-# generate source tarball
58
-git archive --format=tar.gz --output="$DIST_DIR/vllm-${PURE_VERSION}.tar.gz" "$BUILDKITE_COMMIT"
57
+# generate source distribution using setup.py
58
+python setup.py sdist --dist-dir=$DIST_DIR
59
ls -la $DIST_DIR
60
61
+SDIST_FILE=$(find $DIST_DIR -name "vllm*.tar.gz")
62
+echo "Found sdist: $SDIST_FILE"
63
+
64
# upload wheels to PyPI (only default variant, i.e. files without '+' in the name)
65
PYPI_WHEEL_FILES=$(find $DIST_DIR -name "vllm-${PURE_VERSION}*.whl" -not -name "*+*")
66
if [[ -z "$PYPI_WHEEL_FILES" ]]; then
@@ -65,6 +68,6 @@ if [[ -z "$PYPI_WHEEL_FILES" ]]; then
68
exit 1
69
fi
70
68
-python3 -m twine check "$PYPI_WHEEL_FILES"
69
-python3 -m twine upload --non-interactive --verbose "$PYPI_WHEEL_FILES"
70
-echo "Wheels uploaded to PyPI"
71
+python3 -m twine check "$PYPI_WHEEL_FILES" "$SDIST_FILE"
72
+python3 -m twine upload --non-interactive --verbose "$PYPI_WHEEL_FILES" "$SDIST_FILE"
73
+echo "Wheels and source distribution uploaded to PyPI"