Fix exported version of keras's model_to_estimator.

PiperOrigin-RevId: 223944247
2018-12-04 02:29:42 -08:00 · 2018-12-04 02:29:42 -08:00 · 566923394a
commit 566923394a
parent e58a84208a
1 changed files with 1 additions and 1 deletions
--- a/tensorflow/python/keras/estimator/init.py
+++ b/tensorflow/python/keras/estimator/init.py
@ -65,7 +65,7 @@ def model_to_estimator(
    raise NotImplementedError(
        'tf.keras.estimator.model_to_estimator function not available in your '
        'installation.')
-  keras_lib.model_to_estimator(
+  return keras_lib.model_to_estimator(
      keras_model=keras_model,
      keras_model_path=keras_model_path,
      custom_objects=custom_objects,