Fix crash when PYTORCH_VERSION is not defined (pytorch#455)

yanbing-j · Jun 27, 2024 · 1c7c390 · 1c7c390
1 parent 6cab4c8
commit 1c7c390
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packaging/pre_build_script.sh b/packaging/pre_build_script.sh
@@ -10,7 +10,7 @@ set -eux
 echo "This script is run before building torchao binaries"
 
 python -m pip install --upgrade pip
-if [ -z "$PYTORCH_VERSION" ]; then
+if [ -z ${PYTORCH_VERSION:-} ]; then
     PYTORCH_DEP="torch"
 else
     PYTORCH_DEP="torch==$PYTORCH_VERSION"