diff options
Diffstat (limited to 'tensorflow/python/platform/googletest.py')
-rw-r--r-- | tensorflow/python/platform/googletest.py | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/tensorflow/python/platform/googletest.py b/tensorflow/python/platform/googletest.py new file mode 100644 index 0000000000..ca22ec6e6b --- /dev/null +++ b/tensorflow/python/platform/googletest.py @@ -0,0 +1,10 @@ +"""Switch between depending on googletest or unittest.""" +# 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_GOOGLETEST: + from tensorflow.python.platform.default._googletest import * +else: + from tensorflow.python.platform.google._googletest import * |