diff --git a/tensorflow/tools/pip_package/setup.py b/tensorflow/tools/pip_package/setup.py index bb1a49841a8265..79ddc1030710be 100644 --- a/tensorflow/tools/pip_package/setup.py +++ b/tensorflow/tools/pip_package/setup.py @@ -119,7 +119,7 @@ if 'tensorboard' in pkg: REQUIRED_PACKAGES[i] = 'tb-nightly ~= 2.6.0.a' elif 'tensorflow_estimator' in pkg: - REQUIRED_PACKAGES[i] = 'tf-estimator-nightly ~= 2.5.0.dev' + REQUIRED_PACKAGES[i] = 'tf-estimator-nightly ~= 2.6.0.dev' elif 'keras' in pkg and 'keras_preprocessing' not in pkg: REQUIRED_PACKAGES[i] = 'keras-nightly ~= 2.6.0.dev'