Skip to content

Commit 2e23c83

Browse files
vkarampudiResponsible ML Infra Team
authored and
Responsible ML Infra Team
committed
Removing TFDV and TFMA nightly constraints as those are breaking FI&TB nightly's
PiperOrigin-RevId: 714156722
1 parent 70df468 commit 2e23c83

File tree

2 files changed

+3
-18
lines changed

2 files changed

+3
-18
lines changed

setup.py

Lines changed: 2 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -38,18 +38,8 @@ def select_constraint(default, nightly=None, git_master=None):
3838
REQUIRED_PACKAGES = [
3939
'tensorflow>=2.16,<2.17',
4040
'tensorflow-hub>=0.16.1,<1.0.0',
41-
'tensorflow-data-validation'
42-
+ select_constraint(
43-
default='>=1.16.1,<2.0.0',
44-
nightly='>=1.17.0.dev',
45-
git_master='@git+https://github.com/tensorflow/data-validation@master',
46-
),
47-
'tensorflow-model-analysis'
48-
+ select_constraint(
49-
default='>=0.47.0,<0.48.0',
50-
nightly='>=0.48.0.dev',
51-
git_master='@git+https://github.com/tensorflow/model-analysis@master',
52-
),
41+
'tensorflow-data-validation>=1.16.1,<2.0.0',
42+
'tensorflow-model-analysis>=0.47.0,<0.48.0',
5343
'witwidget>=1.4.4,<2',
5444
'protobuf>=3.20.3,<5',
5545
]

tensorboard_plugin/setup.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -45,12 +45,7 @@ def select_constraint(default, nightly=None, git_master=None):
4545
'protobuf>=3.20.3,<5',
4646
'tensorboard>=2.16.2,<2.17.0',
4747
'tensorflow>=2.16,<2.17',
48-
'tensorflow-model-analysis'
49-
+ select_constraint(
50-
default='>=0.47,<0.48',
51-
nightly='>=0.48.0.dev',
52-
git_master='@git+https://github.com/tensorflow/model-analysis@master',
53-
),
48+
'tensorflow-model-analysis>=0.47,<0.48',
5449
'werkzeug<2',
5550
]
5651

0 commit comments

Comments
 (0)