Skip to content

Commit

Permalink
Removing TFDV and TFMA nightly constraints as those are breaking FI&T…
Browse files Browse the repository at this point in the history
…B nightly's

PiperOrigin-RevId: 714156722
  • Loading branch information
vkarampudi authored and Responsible ML Infra Team committed Jan 10, 2025
1 parent 70df468 commit 2e23c83
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 18 deletions.
14 changes: 2 additions & 12 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -38,18 +38,8 @@ def select_constraint(default, nightly=None, git_master=None):
REQUIRED_PACKAGES = [
'tensorflow>=2.16,<2.17',
'tensorflow-hub>=0.16.1,<1.0.0',
'tensorflow-data-validation'
+ select_constraint(
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.47.0,<0.48.0',
nightly='>=0.48.0.dev',
git_master='@git+https://github.com/tensorflow/model-analysis@master',
),
'tensorflow-data-validation>=1.16.1,<2.0.0',
'tensorflow-model-analysis>=0.47.0,<0.48.0',
'witwidget>=1.4.4,<2',
'protobuf>=3.20.3,<5',
]
Expand Down
7 changes: 1 addition & 6 deletions tensorboard_plugin/setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -45,12 +45,7 @@ def select_constraint(default, nightly=None, git_master=None):
'protobuf>=3.20.3,<5',
'tensorboard>=2.16.2,<2.17.0',
'tensorflow>=2.16,<2.17',
'tensorflow-model-analysis'
+ select_constraint(
default='>=0.47,<0.48',
nightly='>=0.48.0.dev',
git_master='@git+https://github.com/tensorflow/model-analysis@master',
),
'tensorflow-model-analysis>=0.47,<0.48',
'werkzeug<2',
]

Expand Down

0 comments on commit 2e23c83

Please sign in to comment.