Do not hardcode 3.10

NVIDIA · May 23, 2024 · a6c3a0e · a6c3a0e
1 parent 9c872f8
commit a6c3a0e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/container/build-jax.sh b/.github/container/build-jax.sh
@@ -275,7 +275,8 @@ line="jaxlib @ file://${BUILD_PATH_JAXLIB}"
 if ! grep -xF "${line}" "${SRC_PATH_JAX}/build/requirements.in"; then
     pushd "${SRC_PATH_JAX}"
     echo "${line}" >> build/requirements.in
-    bazel run //build:requirements_dev.update --repo_env=HERMETIC_PYTHON_VERSION="3.10"
+    PYTHON_VERSION=$(python -c 'import sys; print("{}.{}".format(*sys.version_info[:2]))')
+    bazel run //build:requirements_dev.update --repo_env=HERMETIC_PYTHON_VERSION="${PYTHON_VERSION}"
     popd
 fi