diff options
author | fo40225 <fo40225@hotmail.com> | 2018-07-11 19:09:57 +0800 |
---|---|---|
committer | fo40225 <fo40225@hotmail.com> | 2018-07-11 19:12:58 +0800 |
commit | 53dd73a538916cbfd96877c95dfd42eb6b92ed1c (patch) | |
tree | 22a39dcd95ce42532f289bbf0f8ce3edae6f3b09 | |
parent | b50e7b980c983beab90e62f6013aa3dc2f50ac7f (diff) |
fix cmake python 2.7 test import fail
-rw-r--r-- | tensorflow/__init__.py | 3 | ||||
-rw-r--r-- | tensorflow/contrib/cmake/python_modules.txt | 2 |
2 files changed, 4 insertions, 1 deletions
diff --git a/tensorflow/__init__.py b/tensorflow/__init__.py index 440e9f8dbd..21677512b6 100644 --- a/tensorflow/__init__.py +++ b/tensorflow/__init__.py @@ -28,7 +28,8 @@ contrib = LazyLoader('contrib', globals(), 'tensorflow.contrib') del LazyLoader from tensorflow.python.platform import flags # pylint: disable=g-import-not-at-top -app.flags = flags # pylint: disable=undefined-variable +from tensorflow.python.platform import app # pylint: disable=g-import-not-at-top +app.flags = flags del absolute_import del division diff --git a/tensorflow/contrib/cmake/python_modules.txt b/tensorflow/contrib/cmake/python_modules.txt index 40041d9c88..a465cbf0f1 100644 --- a/tensorflow/contrib/cmake/python_modules.txt +++ b/tensorflow/contrib/cmake/python_modules.txt @@ -4,6 +4,8 @@ tensorflow tensorflow/core tensorflow/core/example tensorflow/core/framework +tensorflow/core/kernels +tensorflow/core/kernels/boosted_trees tensorflow/core/lib tensorflow/core/lib/core tensorflow/core/profiler |