diff --git a/tensorflow/python/keras/distribute/BUILD b/tensorflow/python/keras/distribute/BUILD index f8b260bfb5b..66e2d8a65c8 100644 --- a/tensorflow/python/keras/distribute/BUILD +++ b/tensorflow/python/keras/distribute/BUILD @@ -924,7 +924,6 @@ py_library( "//tensorflow/python:util", "//tensorflow/python:variables", "//tensorflow/python/training/tracking:util", - "@absl_py//absl/logging", ], ) diff --git a/tensorflow/python/keras/distribute/sidecar_evaluator.py b/tensorflow/python/keras/distribute/sidecar_evaluator.py index 1bd2b7e2f48..add90ccd021 100644 --- a/tensorflow/python/keras/distribute/sidecar_evaluator.py +++ b/tensorflow/python/keras/distribute/sidecar_evaluator.py @@ -19,12 +19,12 @@ from __future__ import absolute_import from __future__ import division from __future__ import print_function -from absl import logging # pylint: disable=g-direct-tensorflow-import from tensorflow.python.framework import dtypes from tensorflow.python.framework import errors_impl from tensorflow.python.ops import summary_ops_v2 from tensorflow.python.ops import variables +from tensorflow.python.platform import tf_logging as logging from tensorflow.python.training import checkpoint_utils from tensorflow.python.training.tracking import util as tracking_util