Skip to content

Commit

Permalink
Update keras and estimator deps
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 389672438
Change-Id: Iaee9f29c10caeffad752009faa1fcafab7935b67
  • Loading branch information
mihaimaruseac committed Aug 9, 2021
1 parent 093800c commit 662740b
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions tensorflow/tools/pip_package/setup.py
Expand Up @@ -102,8 +102,8 @@
# They are updated during the release process
# When updating these, please also update the nightly versions below
'tensorboard ~= 2.6',
'tensorflow-estimator >= 2.6.0rc0 , < 2.7.0',
'keras >= 2.6.0rc1 , < 2.7.0',
'tensorflow_estimator ~= 2.6',
'keras ~= 2.6',
]


Expand All @@ -117,9 +117,9 @@
if 'tensorboard' in pkg:
REQUIRED_PACKAGES[i] = 'tb-nightly ~= 2.7.0.a'
elif 'tensorflow_estimator' in pkg:
REQUIRED_PACKAGES[i] = 'tf-estimator-nightly ~= 2.6.0.dev'
REQUIRED_PACKAGES[i] = 'tf-estimator-nightly ~= 2.7.0.dev'
elif 'keras' in pkg and 'keras_preprocessing' not in pkg:
REQUIRED_PACKAGES[i] = 'keras-nightly ~= 2.6.0.dev'
REQUIRED_PACKAGES[i] = 'keras-nightly ~= 2.7.0.dev'


# grpcio does not build correctly on big-endian machines due to lack of
Expand Down

0 comments on commit 662740b

Please sign in to comment.