From 43334d889e026ab5c199980beeba698b59e649ed Mon Sep 17 00:00:00 2001
From: Richard Stotz <richardstotz@google.com>
Date: Tue, 27 Feb 2024 01:04:09 -0800
Subject: [PATCH] Prepare release of TF-DF 1.9.0rc0 and YDF 1.9.0rc0

PiperOrigin-RevId: 610665881
---
 CHANGELOG.md                            | 13 +++++++++++++
 configure/setup.py                      |  4 ++--
 tensorflow_decision_forests/__init__.py |  4 ++--
 tools/start_compile_docker.sh           |  2 +-
 tools/test_bazel.sh                     |  2 +-
 5 files changed, 19 insertions(+), 6 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index eba58d6..48008c6 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -1,5 +1,18 @@
 # Changelog
 
+## 1.9.0rc0 - 2024-02-26
+
+### Fix
+
+-   Fix max_depth, early stopping parameter documentation.
+-   Fix plotting contains conditions.
+
+### Features
+
+-   Compatibility with TensorFlow 2.16.0rc0.
+-   Using tf_keras instead tf.keras in examples, documentation.
+-   Support NAConditions for fast engine.
+
 ## 1.8.1 - 2023-11-17
 
 ### Fix
diff --git a/configure/setup.py b/configure/setup.py
index c7e0bb6..f43bb8d 100644
--- a/configure/setup.py
+++ b/configure/setup.py
@@ -21,7 +21,7 @@
 from setuptools.command.install import install
 from setuptools.dist import Distribution
 
-_VERSION = "1.8.1"
+_VERSION = "1.9.0rc0"
 
 with open("README.md", "r", encoding="utf-8") as fh:
   long_description = fh.read()
@@ -29,7 +29,7 @@
 REQUIRED_PACKAGES = [
     "numpy",
     "pandas",
-    "tensorflow~=2.15.0",
+    "tensorflow~=2.16.0rc0",
     "six",
     "absl_py",
     "wheel",
diff --git a/tensorflow_decision_forests/__init__.py b/tensorflow_decision_forests/__init__.py
index 5da90f9..5ec9ac4 100644
--- a/tensorflow_decision_forests/__init__.py
+++ b/tensorflow_decision_forests/__init__.py
@@ -51,10 +51,10 @@
 
 """
 
-__version__ = "1.8.1"
+__version__ = "1.9.0rc0"
 __author__ = "Mathieu Guillame-Bert"
 
-compatible_tf_versions = ["2.15.0"]
+compatible_tf_versions = ["2.16.0rc0"]
 __git_version__ = "HEAD"  # Modify for release build.
 
 from tensorflow_decision_forests.tensorflow import check_version
diff --git a/tools/start_compile_docker.sh b/tools/start_compile_docker.sh
index 2801a69..8710160 100755
--- a/tools/start_compile_docker.sh
+++ b/tools/start_compile_docker.sh
@@ -58,7 +58,7 @@
 #  ./tools/build_pip_package.sh ALL_VERSIONS_ALREADY_ASSEMBLED
 #
 # https://hub.docker.com/r/tensorflow/build/tags?page=1
-DOCKER=tensorflow/build:2.15-python3.9
+DOCKER=tensorflow/build:2.16-python3.9
 
 # Current directory
 # Useful if Yggdrasil Decision Forests is available locally in a neighbor
diff --git a/tools/test_bazel.sh b/tools/test_bazel.sh
index 547d48e..c13a1bf 100755
--- a/tools/test_bazel.sh
+++ b/tools/test_bazel.sh
@@ -28,7 +28,7 @@
 #
 # Usage example
 #
-#   RUN_TESTS=1 PY_VERSION=3.9 TF_VERSION=2.13.0 ./tools/test_bazel.sh
+#   RUN_TESTS=1 PY_VERSION=3.9 TF_VERSION=2.15.0 ./tools/test_bazel.sh
 
 set -vex