diff --git a/tfx/dependencies.py b/tfx/dependencies.py index f6c12f9325..2ac84e5b62 100644 --- a/tfx/dependencies.py +++ b/tfx/dependencies.py @@ -106,11 +106,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( - default='>=2.15.0,<2.16', - nightly='>=2.16.0.dev', - git_master='@git+https://github.com/tensorflow/tensorflow@master', - ), + 'tensorflow' + select_constraint('>=2.15.0,<2.16'), # pylint: enable=line-too-long 'tensorflow-hub>=0.15.0,<0.16', 'tensorflow-data-validation'