Fix model to estimator breakage caused by recent change.

PiperOrigin-RevId: 380894633

Fix model to estimator breakage caused by recent change.
PiperOrigin-RevId: 380894633
16608a98 · Scott Zhu · TensorFlower Gardener · 48141423 · 16608a98
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

keras/estimator/__init__.py keras/estimator/__init__.py +1 -1

未找到文件。
--- a/keras/estimator/__init__.py
+++ b/keras/estimator/__init__.py
@@ -160,7 +160,7 @@ def model_to_estimator(
  """

  try:
-    from tensorflow_estimator.python.estimator import keras as keras_lib  # pylint: disable=g-import-not-at-top
+    from tensorflow_estimator.python.estimator import keras_lib  # pylint: disable=g-import-not-at-top
  except ImportError:
    raise NotImplementedError(
        'tf.keras.estimator.model_to_estimator function not available in your '