fix nightly package installation for tf.text ci jobs

PiperOrigin-RevId: 641960415
tensorflow · Jun 10, 2024 · 7e2412c · 7e2412c
1 parent 0fa2435
commit 7e2412c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 10 deletions.
diff --git a/oss_scripts/configure.sh b/oss_scripts/configure.sh
@@ -68,6 +68,17 @@ 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])))")
+export HERMETIC_PYTHON_VERSION
+
+echo "TF_VERSION=$TF_VERSION"
+REQUIREMENTS_EXTRA_FLAGS="--upgrade"
+if [[ "$TF_VERSION" == *"rc"* ]]; then
+  REQUIREMENTS_EXTRA_FLAGS="$REQUIREMENTS_EXTRA_FLAGS --pre"
+fi
+
+bazel run //oss_scripts/pip_package:requirements.update -- $REQUIREMENTS_EXTRA_FLAGS
+
 TF_ABIFLAG=$(bazel run //oss_scripts/pip_package:tensorflow_build_info -- abi)
 
 HEADER_DIR=${TF_CFLAGS:2}

diff --git a/oss_scripts/prepare_tf_dep.sh b/oss_scripts/prepare_tf_dep.sh
@@ -13,16 +13,6 @@ if [[ "${osname}" == "darwin" ]]; then
   ext='""'
 fi
 
-HERMETIC_PYTHON_VERSION=$($installed_python  -c "import sys; print('.'.join(map(str, sys.version_info[:2])))")
-export HERMETIC_PYTHON_VERSION
-
-
-echo "TF_VERSION=$TF_VERSION"
-REQUIREMENTS_EXTRA_FLAGS="--upgrade"
-if [[ "$TF_VERSION" == *"rc"* ]]; then
-  REQUIREMENTS_EXTRA_FLAGS="$REQUIREMENTS_EXTRA_FLAGS --pre"
-fi
-
 bazel run //oss_scripts/pip_package:requirements.update -- $REQUIREMENTS_EXTRA_FLAGS
 
 # Update setup.nightly.py with current tf version.