commit 7c651c3ea2c1255a07f490bae9b43470e85d8645
author Mihai Maruseac <mihaimaruseac@google.com> Fri Oct 16 16:41:47 2020 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> Fri Oct 16 16:49:30 2020 -0700
tree 68b6f3a541503f4db01d46293c321a6a3b14ee54
parent b296b9f1fd5e245afe60e480fe9478601cf39e10

Fix setup.py handling of estimator nightly.

PiperOrigin-RevId: 337600381
Change-Id: I24ee3b6dd6c57effc189ef5fb5574e349b68aded

tensorflow/tools/pip_package/setup.py

diff --git a/tensorflow/tools/pip_package/setup.py b/tensorflow/tools/pip_package/setup.py
index 0dc0391..d68422c 100644
--- a/tensorflow/tools/pip_package/setup.py
+++ b/tensorflow/tools/pip_package/setup.py
@@ -108,7 +108,10 @@
     if 'tensorboard' in pkg:
       REQUIRED_PACKAGES[i] = 'tb-nightly ~= 2.4'
     elif 'tensorflow_estimator' in pkg:
-      REQUIRED_PACKAGES[i] = 'tf-estimator-nightly ~= 2.4.0'
+      # Note the .* part! TF Estimator nightly packages are not named in a way
+      # that makes `~=` syntax work so we use `==` and `.*` to pick the latest
+      # released nightly.
+      REQUIRED_PACKAGES[i] = 'tf-estimator-nightly == 2.4.0.*'
 
 
 # grpcio does not build correctly on big-endian machines due to lack of