diff --git a/tensorflow/python/util/lazy_loader.py b/tensorflow/python/util/lazy_loader.py index a9499f8334c518..e237c844e8914f 100644 --- a/tensorflow/python/util/lazy_loader.py +++ b/tensorflow/python/util/lazy_loader.py @@ -21,6 +21,7 @@ import importlib import types +from tensorflow.python.platform import tf_logging as logging class LazyLoader(types.ModuleType): @@ -46,7 +47,7 @@ def _load(self): # Emit a warning if one was specified if self._warning: - print(self._warning) + logging.warning(self._warning) # Make sure to only warn once. self._warning = None