From a6f6784dd2fc5e7915c944cb2c3d1716808fd588 Mon Sep 17 00:00:00 2001
From: Jacob Burnim <jburnim@google.com>
Date: Tue, 12 Mar 2024 17:10:30 +0000
Subject: [PATCH] Update version and TF requirement for TFP 0.24.0 release.

---
 setup.py                                  | 4 ++--
 tensorflow_probability/python/__init__.py | 2 +-
 tensorflow_probability/python/version.py  | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/setup.py b/setup.py
index 9768d8f238..7b6ffd83a7 100644
--- a/setup.py
+++ b/setup.py
@@ -49,8 +49,8 @@
   TFDS_PACKAGE = 'tfds-nightly'
 
 if release:
-  TF_PACKAGE = 'tensorflow >= 2.15'
-  KERAS_PACKAGE = 'tf-keras >= 2.15'
+  TF_PACKAGE = 'tensorflow >= 2.16'
+  KERAS_PACKAGE = 'tf-keras >= 2.16'
 else:
   TF_PACKAGE = 'tf-nightly'
   KERAS_PACKAGE = 'tf-keras-nightly'
diff --git a/tensorflow_probability/python/__init__.py b/tensorflow_probability/python/__init__.py
index dce1163150..f31a40384f 100644
--- a/tensorflow_probability/python/__init__.py
+++ b/tensorflow_probability/python/__init__.py
@@ -51,7 +51,7 @@ def _validate_tf_environment(package):
   #
   # Update this whenever we need to depend on a newer TensorFlow release.
   #
-  required_tensorflow_version = '2.15'
+  required_tensorflow_version = '2.16'
 #   required_tensorflow_version = '1.15'  # Needed internally -- DisableOnExport
 
   if (distutils.version.LooseVersion(tf.__version__) <
diff --git a/tensorflow_probability/python/version.py b/tensorflow_probability/python/version.py
index 2cce99e09e..c93931ddc7 100644
--- a/tensorflow_probability/python/version.py
+++ b/tensorflow_probability/python/version.py
@@ -24,7 +24,7 @@
 # stable release (indicated by `_VERSION_SUFFIX = ''`). Outside the context of a
 # release branch, the current version is by default assumed to be a
 # 'development' version, labeled 'dev'.
-_VERSION_SUFFIX = 'dev'
+_VERSION_SUFFIX = ''
 
 # Example, '0.4.0-dev'
 __version__ = '.'.join([