Merge branch 'r2.7' into r2.7

tensorflow · Oct 13, 2021 · d1f5f0b · d1f5f0b
2 parents f8691f9 + cfe7e5c
commit d1f5f0b
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/tensorflow/tools/pip_package/setup.py b/tensorflow/tools/pip_package/setup.py
@@ -102,8 +102,10 @@
     # They are updated during the release process
     # When updating these, please also update the nightly versions below
     'tensorboard ~= 2.6',
-    'tensorflow_estimator ~= 2.6',
-    'keras ~= 2.6',
+    'tensorflow_estimator ~= 2.7.0rc0, < 2.8',
+    # Keras release is not backward compatible with old tf release, and we have
+    # to make the version aligned between TF and Keras.
+    'keras >= 2.7.0rc0, < 2.8',
     'tensorflow-io-gcs-filesystem >= 0.21.0',
 ]
 
@@ -118,9 +120,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.7.0.dev'
+      REQUIRED_PACKAGES[i] = 'tf-estimator-nightly ~= 2.8.0.dev'
     elif 'keras' in pkg and 'keras_preprocessing' not in pkg:
-      REQUIRED_PACKAGES[i] = 'keras-nightly ~= 2.7.0.dev'
+      REQUIRED_PACKAGES[i] = 'keras-nightly ~= 2.8.0.dev'
 
 
 # grpcio does not build correctly on big-endian machines due to lack of