diff --git a/RELEASE.md b/RELEASE.md index 85d6828..4e7d8b7 100644 --- a/RELEASE.md +++ b/RELEASE.md @@ -8,7 +8,7 @@ * Bumped the Ubuntu version on which `struct2tensor` is tested to 20.04 (previously was 16.04). -* Depends on `tensorflow>=2.15.0,<3`. +* Depends on `tensorflow~=2.15.0`. * Bumped the minimum bazel version required to build `struct2tensor` to 6.1.0. * Depends on `protobuf>=4.25.2,<5` for Python 3.11 and on `protobuf>3.20.3,<5` for 3.9 and 3.10. diff --git a/setup.py b/setup.py index f3bcbd3..61b3aa8 100644 --- a/setup.py +++ b/setup.py @@ -79,7 +79,7 @@ def select_constraint(default, nightly=None, git_master=None): 'numpy>=1.22', 'protobuf>=4.25.2,<5;python_version>="3.11"', 'protobuf>=3.20.3,<5;python_version<"3.11"', - 'tensorflow>=2.15,<3', + 'tensorflow~=2.15', 'tensorflow-metadata' + select_constraint( default='>=1.14.0,<1.15.0', nightly='>=1.15.0.dev',