diff --git a/tensorflow/tools/pip_package/setup.py b/tensorflow/tools/pip_package/setup.py index c62b237f1ba..328ac2a988f 100644 --- a/tensorflow/tools/pip_package/setup.py +++ b/tensorflow/tools/pip_package/setup.py @@ -64,7 +64,7 @@ REQUIRED_PACKAGES = [ 'opt_einsum >= 2.3.2', 'protobuf >= 3.9.2', 'tensorboard >= 2.2.0, < 2.3.0', - 'tf-estimator-nightly >= 2.3.0.dev2020062301, < 2.3.0.dev2020062302', + 'tensorflow_estimator >= 2.3.0-rc0, < 2.4.0', 'termcolor >= 1.1.0', 'wrapt >= 1.11.1', 'wheel >= 0.26',