Skip to content

Commit 7438b3c

Browse files
committed
update setup.py to current build
1 parent 5f41842 commit 7438b3c

File tree

1 file changed

+14
-12
lines changed

1 file changed

+14
-12
lines changed

setup.py

Lines changed: 14 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -182,23 +182,24 @@ def select_constraint(default, nightly=None, git_master=None):
182182
'joblib>=1.2.0', # Dependency for multi-processing.
183183
'numpy>=1.22.0',
184184
'pandas>=1.0,<2',
185-
'protobuf>=4.25.2,<5;python_version>="3.11"',
185+
'protobuf>=4.25.2,<6;python_version>="3.11"',
186186
'protobuf>=3.20.3,<5;python_version<"3.11"',
187187
'pyarrow>=10,<11',
188188
'pyfarmhash>=0.2.2,<0.4',
189189
'six>=1.12,<2',
190-
'tensorflow' + select_constraint(
191-
default='>=2.16,<2.17',
192-
nightly='>=2.17.0.dev',
193-
git_master='@git+https://github.com/tensorflow/tensorflow@master'),
194-
'tensorflow-metadata' + select_constraint(
195-
default='>=1.16.0,<1.17',
190+
'tensorflow>=2.17,<2.18',
191+
'tensorflow-metadata'
192+
+ select_constraint(
193+
default='>=1.16.1,<1.17',
196194
nightly='>=1.17.0.dev',
197-
git_master='@git+https://github.com/tensorflow/metadata@master'),
198-
'tfx-bsl' + select_constraint(
199-
default='>=1.16.0,<1.17',
195+
git_master='@git+https://github.com/tensorflow/metadata@master',
196+
),
197+
'tfx-bsl'
198+
+ select_constraint(
199+
default='>=1.16.1,<1.17',
200200
nightly='>=1.17.0.dev',
201-
git_master='@git+https://github.com/tensorflow/tfx-bsl@master'),
201+
git_master='@git+https://github.com/tensorflow/tfx-bsl@master',
202+
),
202203
],
203204
extras_require={
204205
'mutual-information': _make_mutual_information_requirements(),
@@ -222,4 +223,5 @@ def select_constraint(default, nightly=None, git_master=None):
222223
'install': _InstallPlatlibCommand,
223224
'build': _BuildCommand,
224225
'bazel_build': _BazelBuildCommand,
225-
})
226+
},
227+
)

0 commit comments

Comments
 (0)