Skip to content

Commit

Permalink
Update required packages for fairness indicators to TF 2.16, TFMA 0.4…
Browse files Browse the repository at this point in the history
…7.0.dev, and TFDV 1.16.1.

PiperOrigin-RevId: 689010533
  • Loading branch information
vkarampudi authored and Responsible ML Infra Team committed Nov 6, 2024
1 parent 1545d46 commit 54e0ffa
Showing 1 changed file with 4 additions and 5 deletions.
9 changes: 4 additions & 5 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,15 +38,14 @@ def select_constraint(default, nightly=None, git_master=None):
return default

REQUIRED_PACKAGES = [
'tensorflow>=2.15,<2.16',
'tensorflow>=2.16,<2.17',
'tensorflow-hub>=0.16.1,<1.0.0',
'tensorflow-data-validation' + select_constraint(
default='>=1.15.1,<2.0.0',
nightly='>=1.16.0.dev',
default='>=1.16.1,<2.0.0',
nightly='>=1.17.0.dev',
git_master='@git+https://github.com/tensorflow/data-validation@master'),
'tensorflow-model-analysis' + select_constraint(
default='>=0.46,<0.47',
nightly='>=0.47.0.dev',
default='>=0.47.0.dev',
git_master='@git+https://github.com/tensorflow/model-analysis@master'),
'witwidget>=1.4.4,<2',
'protobuf>=3.20.3,<5',
Expand Down

0 comments on commit 54e0ffa

Please sign in to comment.