Update dependency requirements for tensorflow and version.

PiperOrigin-RevId: 616924984
tensorflow · Mar 18, 2024 · 5b1d133 · 5b1d133
1 parent 28a0a80
commit 5b1d133
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 3 deletions.
diff --git a/docs/api_docs/python/tfr.md b/docs/api_docs/python/tfr.md
@@ -42,7 +42,7 @@ TensorFlow Ranking, the project to build ranking models on TensorFlow.
 **version**<a id="__version__"></a>
 </td>
 <td>
-`'0.5.4.dev'`
+`'0.5.5.dev'`
 </td>
 </tr>
 </table>
diff --git a/docs/api_docs/python/tfr/api_report.pb b/docs/api_docs/python/tfr/api_report.pb
diff --git a/tensorflow_ranking/python/version.py b/tensorflow_ranking/python/version.py
@@ -17,7 +17,7 @@
 # We follow Semantic Versioning (https://semver.org/)
 _MAJOR_VERSION = '0'
 _MINOR_VERSION = '5'
-_PATCH_VERSION = '4'
+_PATCH_VERSION = '5'
 
 # When building releases, we can update this value on the release branch to
 # reflect the current release candidate ('rc0', 'rc1') or, finally, the official

diff --git a/tensorflow_ranking/tools/pip_package/setup.py b/tensorflow_ranking/tools/pip_package/setup.py
@@ -28,13 +28,14 @@
 # This version string is semver compatible, but incompatible with pip.
 # For pip, we will remove all '-' characters from this string, and use the
 # result for pip.
-_VERSION = '0.5.4'
+_VERSION = '0.5.5'
 
 REQUIRED_PACKAGES = [
     'absl-py >= 0.1.6',
     'numpy >= 1.23.2',
     'six >= 1.10.0',
     'tensorflow-serving-api >= 2.0.0, < 3.0.0',
+    'tensorflow < 2.16.0',
 ]
 
 project_name = 'tensorflow_ranking'