From 126081204f6095ea9ebbffd470b5ed961bf79b29 Mon Sep 17 00:00:00 2001 From: tf-transform-team Date: Tue, 30 Apr 2024 13:11:30 -0700 Subject: [PATCH] Allow dev and github main versions to depend on corresponding Tensorflow dev+main versions. PiperOrigin-RevId: 629506459 --- setup.py | 19 ++++++++++++++----- 1 file changed, 14 insertions(+), 5 deletions(-) diff --git a/setup.py b/setup.py index 73b38f9..ddcc626 100644 --- a/setup.py +++ b/setup.py @@ -50,15 +50,24 @@ def _make_required_install_packages(): 'protobuf>=3.20.3,<5;python_version<"3.11"', 'pyarrow>=10,<11', 'pydot>=1.2,<2', - 'tensorflow>=2.15,<2.16', - 'tensorflow-metadata' + select_constraint( + 'tensorflow' + + select_constraint( + default='>=2.15,<2.16', + nightly='>=2.16.0.dev', + git_master='@git+https://github.com/tensorflow/tensorflow@master', + ), + 'tensorflow-metadata' + + select_constraint( default='>=1.15.0,<1.16.0', nightly='>=1.16.0.dev', - git_master='@git+https://github.com/tensorflow/metadata@master'), - 'tfx-bsl' + select_constraint( + git_master='@git+https://github.com/tensorflow/metadata@master', + ), + 'tfx-bsl' + + select_constraint( default='>=1.15.1,<1.16.0', nightly='>=1.16.0.dev', - git_master='@git+https://github.com/tensorflow/tfx-bsl@master'), + git_master='@git+https://github.com/tensorflow/tfx-bsl@master', + ), ]