diff --git a/setup.py b/setup.py index 90ef01e..76e0832 100644 --- a/setup.py +++ b/setup.py @@ -38,7 +38,7 @@ def select_constraint(default, nightly=None, git_master=None): return default REQUIRED_PACKAGES = [ - 'tensorflow>=2.15,<3', + 'tensorflow~=2.15', 'tensorflow-hub>=0.16.1,<1.0.0', 'tensorflow-data-validation' + select_constraint( default='>=1.14.0,<1.15.0', diff --git a/tensorboard_plugin/setup.py b/tensorboard_plugin/setup.py index 2c1c50b..d6a9542 100644 --- a/tensorboard_plugin/setup.py +++ b/tensorboard_plugin/setup.py @@ -44,7 +44,7 @@ def select_constraint(default, nightly=None, git_master=None): REQUIRED_PACKAGES = [ 'protobuf>=3.20.3,<5', 'tensorboard>=2.15.2,<2.16.0', - 'tensorflow>=2.15,<3', + 'tensorflow~=2.15', 'tensorflow-model-analysis' + select_constraint( default='>=0.45,<=0.46',