Skip to content

Commit 1260812

Browse files
tf-transform-teamtfx-copybara
authored andcommitted
Allow dev and github main versions to depend on corresponding Tensorflow
dev+main versions. PiperOrigin-RevId: 629506459
1 parent 4c7f784 commit 1260812

File tree

1 file changed

+14
-5
lines changed

1 file changed

+14
-5
lines changed

setup.py

Lines changed: 14 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -50,15 +50,24 @@ def _make_required_install_packages():
5050
'protobuf>=3.20.3,<5;python_version<"3.11"',
5151
'pyarrow>=10,<11',
5252
'pydot>=1.2,<2',
53-
'tensorflow>=2.15,<2.16',
54-
'tensorflow-metadata' + select_constraint(
53+
'tensorflow'
54+
+ select_constraint(
55+
default='>=2.15,<2.16',
56+
nightly='>=2.16.0.dev',
57+
git_master='@git+https://github.com/tensorflow/tensorflow@master',
58+
),
59+
'tensorflow-metadata'
60+
+ select_constraint(
5561
default='>=1.15.0,<1.16.0',
5662
nightly='>=1.16.0.dev',
57-
git_master='@git+https://github.com/tensorflow/metadata@master'),
58-
'tfx-bsl' + select_constraint(
63+
git_master='@git+https://github.com/tensorflow/metadata@master',
64+
),
65+
'tfx-bsl'
66+
+ select_constraint(
5967
default='>=1.15.1,<1.16.0',
6068
nightly='>=1.16.0.dev',
61-
git_master='@git+https://github.com/tensorflow/tfx-bsl@master'),
69+
git_master='@git+https://github.com/tensorflow/tfx-bsl@master',
70+
),
6271
]
6372

6473

0 commit comments

Comments
 (0)