blob: b6d2e53dd4b5ff1004ac8e16b4eb3e17b29a56fb (
plain)
1
2
3
4
5
6
7
8
9
10
|
"""Switch between depending on pyglib.logging or regular logging."""
# pylint: disable=unused-import
# pylint: disable=g-import-not-at-top
# pylint: disable=wildcard-import
import tensorflow.python.platform
import control_imports
if control_imports.USE_OSS and control_imports.OSS_LOGGING:
from tensorflow.python.platform.default._logging import *
else:
from tensorflow.python.platform.google._logging import *
|
