Set HERMETIC_PYTHON_VERSION correctly

PiperOrigin-RevId: 706847683
tensorflow · Dec 16, 2024 · 966d976 · 966d976
1 parent 534859c
commit 966d976
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/oss_scripts/configure.sh b/oss_scripts/configure.sh
@@ -61,7 +61,13 @@ elif (which python) | grep -q "python"; then
   installed_python="python"
 fi
 
-HERMETIC_PYTHON_VERSION=$($installed_python  -c "import sys; print('.'.join(map(str, sys.version_info[:2])))")
+if [ -z "$HERMETIC_PYTHON_VERSION" ]; then
+  if [ -n "$PY_VERSION" ]; then
+    HERMETIC_PYTHON_VERSION=$PY_VERSION
+  else
+    HERMETIC_PYTHON_VERSION=$($installed_python  -c "import sys; print('.'.join(map(str, sys.version_info[:2])))")
+  fi
+fi
 export HERMETIC_PYTHON_VERSION
 
 echo "TF_VERSION=$TF_VERSION"