Allow pre-release

This commit is contained in:
Lucas Armand
2026-03-24 13:09:04 -07:00
parent 50633c5003
commit 62cd96ea68
+3 -3
View File
@@ -53,7 +53,7 @@ function install_vastai_sdk() {
echo "WARNING: Both SDK_BRANCH and SDK_VERSION are set; using SDK_BRANCH=${SDK_BRANCH}"
fi
echo "Installing vastai-sdk from https://github.com/vast-ai/vast-sdk/ @ ${SDK_BRANCH}"
if ! uv pip install "vastai-sdk @ git+https://github.com/vast-ai/vast-sdk.git@${SDK_BRANCH}"; then
if ! uv pip install --prerelease=allow "vastai-sdk @ git+https://github.com/vast-ai/vast-sdk.git@${SDK_BRANCH}"; then
report_error_and_exit "Failed to install vastai-sdk from vast-ai/vast-sdk@${SDK_BRANCH}"
fi
return 0
@@ -61,14 +61,14 @@ function install_vastai_sdk() {
if [ -n "${SDK_VERSION:-}" ]; then
echo "Installing vastai-sdk version ${SDK_VERSION}"
if ! uv pip install "vastai-sdk==${SDK_VERSION}"; then
if ! uv pip install --prerelease=allow "vastai-sdk==${SDK_VERSION}"; then
report_error_and_exit "Failed to install vastai-sdk==${SDK_VERSION}"
fi
return 0
fi
echo "Installing default vastai-sdk"
if ! uv pip install vastai-sdk; then
if ! uv pip install --prerelease=allow vastai-sdk; then
report_error_and_exit "Failed to install vastai-sdk"
fi
}