From da845a4a531a50b19fbf3139405c19694ff28eb9 Mon Sep 17 00:00:00 2001 From: vkarampudi Date: Tue, 23 Jul 2024 11:55:03 -0700 Subject: [PATCH] Internal change PiperOrigin-RevId: 655250076 --- test_constraints.txt | 6 +----- tfx/dependencies.py | 2 +- 2 files changed, 2 insertions(+), 6 deletions(-) diff --git a/test_constraints.txt b/test_constraints.txt index 131727aa28..4caa86f224 100644 --- a/test_constraints.txt +++ b/test_constraints.txt @@ -9,8 +9,4 @@ # dependencies. # TODO(b/321609768): Remove pinned Flask-session version after resolving the issue. -Flask-session<0.6.0 - -#TODO(b/329181965): Remove once we migrate TFX to 2.16. -tensorflow<2.16 -tensorflow-text<2.16 \ No newline at end of file +Flask-session<0.6.0 \ No newline at end of file diff --git a/tfx/dependencies.py b/tfx/dependencies.py index 84098ae7c0..55040ac855 100644 --- a/tfx/dependencies.py +++ b/tfx/dependencies.py @@ -100,7 +100,7 @@ def make_required_install_packages(): # Pip might stuck in a TF 1.15 dependency although there is a working # dependency set with TF 2.x without the sync. # pylint: disable=line-too-long - 'tensorflow' + select_constraint('>=2.15.0,<2.16'), + 'tensorflow' + select_constraint('>=2.16.0,<2.17'), # pylint: enable=line-too-long 'tensorflow-hub>=0.15.0,<0.16', 'tensorflow-data-validation'