1.9.0rc0 fixes

PiperOrigin-RevId: 612435124

Author: rstz

configure/setup.py

@@ -34,7 +34,7 @@
     "absl_py",
     "wheel",
     "wurlitzer",
-    "tf_keras",
+    "tf_keras~=2.16.0rc2",
 ]
 
 

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

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=""