提交 88cabd5c 编写于 作者: S Scott Zhu 提交者: TensorFlower Gardener

Rename the keras.py to keras_lib.py to resolve the name conflict during OSS test.

PiperOrigin-RevId: 380851544
上级 91eb589f
......@@ -348,7 +348,7 @@ def model_to_estimator_v2(keras_model=None,
"""
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 '
......@@ -364,4 +364,4 @@ def model_to_estimator_v2(keras_model=None,
use_v2_estimator=True,
metric_names_map=metric_names_map,
export_outputs=export_outputs)
# LINT.ThenChange(//tensorflow_estimator/python/estimator/keras.py)
# LINT.ThenChange(//tensorflow_estimator/python/estimator/keras_lib.py)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册