From 63ecbd69b30d38a0ff9b4fb26d0ecdfe153162d4 Mon Sep 17 00:00:00 2001 From: tfx-bsl-team Date: Thu, 7 Dec 2023 14:13:31 -0800 Subject: [PATCH] Updating Tensorflow dependency to `>=2.15,<3`. PiperOrigin-RevId: 588906079 --- RELEASE.md | 3 ++- WORKSPACE | 4 ++-- setup.py | 2 +- 3 files changed, 5 insertions(+), 4 deletions(-) diff --git a/RELEASE.md b/RELEASE.md index 4df7676..1be9404 100644 --- a/RELEASE.md +++ b/RELEASE.md @@ -8,7 +8,8 @@ * Bump the macOS version on which TFX-BSL is tested to Ventura (previously was Monterey). -* Bumps the pybind11 version to 2.11.1 +* Bumps the pybind11 version to 2.11.1 +* Depends on `tensorflow>=2.15,<3` ## Breaking Changes diff --git a/WORKSPACE b/WORKSPACE index b141e50..51be0f7 100644 --- a/WORKSPACE +++ b/WORKSPACE @@ -89,10 +89,10 @@ http_archive( # TODO(b/177694034): Follow the new format for tensorflow import after TF 2.5. #here -TENSORFLOW_COMMIT = "1cb1a030a62b169d90d34c747ab9b09f332bf905" # 2.13.0 +TENSORFLOW_COMMIT = "6887368d6d46223f460358323c4b76d61d1558a8" # 2.15.0 http_archive( name = "org_tensorflow_no_deps", - sha256 = "a62eba23ebfcf1d6d2d3241f1629b99df576a9f726c439a97c3acd590e71fe62", + sha256 = "bb25fa4574e42ea4d452979e1d2ba3b86b39569d6b8106a846a238b880d73652", strip_prefix = "tensorflow-%s" % TENSORFLOW_COMMIT, urls = [ "https://mirror.bazel.build/github.com/tensorflow/tensorflow/archive/%s.tar.gz" % TENSORFLOW_COMMIT, diff --git a/setup.py b/setup.py index a505d33..54e2702 100644 --- a/setup.py +++ b/setup.py @@ -174,7 +174,7 @@ def select_constraint(default, nightly=None, git_master=None): 'pandas>=1.0,<2', 'protobuf>=3.20.3,<5', 'pyarrow>=10,<11', - 'tensorflow>=2.13,<3', + 'tensorflow>=2.15,<3', 'tensorflow-metadata' + select_constraint( default='>=1.14.0,<1.15.0', nightly='>=1.15.0.dev',