1.9.0rc0 fixes

PiperOrigin-RevId: 612435124
tensorflow · Mar 4, 2024 · f2ed042 · f2ed042
1 parent 43334d8
commit f2ed042
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 4 deletions.
diff --git a/configure/setup.py b/configure/setup.py
@@ -34,7 +34,7 @@
     "absl_py",
     "wheel",
     "wurlitzer",
-    "tf_keras",
+    "tf_keras~=2.16.0rc2",
 ]
 
 

diff --git a/tensorflow_decision_forests/__init__.py b/tensorflow_decision_forests/__init__.py
@@ -54,7 +54,7 @@
 __version__ = "1.9.0rc0"
 __author__ = "Mathieu Guillame-Bert"
 
-compatible_tf_versions = ["2.16.0rc0"]
+compatible_tf_versions = ["2.16.0-rc0"]
 __git_version__ = "HEAD"  # Modify for release build.
 
 from tensorflow_decision_forests.tensorflow import check_version

diff --git a/tools/test_bazel.sh b/tools/test_bazel.sh
@@ -59,8 +59,12 @@ if [ ${TF_VERSION} == "nightly" ]; then
   TF_MINOR="nightly"
 else
   ${PYTHON} -m pip install tensorflow==${TF_VERSION} --force-reinstall
-  TF_MINOR=$(echo $TF_VERSION | grep -oP '[0-9]+\.[0-9]+')
-  ${PYTHON} -m pip install tf-keras==${TF_MINOR}  --force-reinstall
+  if [[ $TF_VERSION == *"rc"* ]]; then
+    ${PYTHON} -m pip install tf-keras --pre --upgrade
+  else
+    TF_MINOR=$(echo $TF_VERSION | grep -oP '[0-9]+\.[0-9]+')
+    ${PYTHON} -m pip install tf-keras==${TF_MINOR}
+  fi
 fi
 ext=""