Update setup.py

This commit is contained in:
Mihai Maruseac 2021-03-25 18:02:39 -07:00 committed by GitHub
parent b42047d765
commit b84dac5ea8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -101,7 +101,7 @@ REQUIRED_PACKAGES = [
# They are updated during the release process
# When updating these, please also update the nightly versions below
'tensorboard ~= 2.4',
'tensorflow_estimator ~= 2.4.0',
'tf-estimator-nightly == 2.5.0.dev2021032501`,
# TODO(scottzhu): OSS keras hasn't been formally released yet.
# Use keras-nightly at the moment.
'keras-nightly ~= 2.5.0.dev',