From 1909ac9b00c30b41acb6719445cc72a81eb63a1d Mon Sep 17 00:00:00 2001 From: Mark Daoust Date: Fri, 18 Jun 2021 13:34:43 -0700 Subject: [PATCH] Fix `tf.keras.estimator.model_to_estimator` in `tf_nightly`. With the keras implementation moved the and the old estimator version here, model_to_estimator was reading the class definitions from the tensorflow source instead of the keras source, so some `isinstance()` checks were comparing two different implementations of the same class and returning False when it should have been True. PiperOrigin-RevId: 380254729 Change-Id: If506ccdac39e9aacd9bc38b14608aaaf287e98a4 --- tensorflow/tools/pip_package/setup.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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'