diff options
author | 2018-05-31 22:55:46 -0700 | |
---|---|---|
committer | 2018-05-31 22:55:46 -0700 | |
commit | 8f79ab773fe44e4779138a77a3bda4b18245d658 (patch) | |
tree | 26b0d57fda4d9ac47f3bf53c681c1d97ff6c2f0a /tensorflow/contrib/lite/python | |
parent | 1acaca5c2b033f2d51f7d2e97da0511b04420f1d (diff) |
Fix import depth issue.
Diffstat (limited to 'tensorflow/contrib/lite/python')
-rw-r--r-- | tensorflow/contrib/lite/python/lite.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/python/lite.py b/tensorflow/contrib/lite/python/lite.py index 0fc7958d41..d595415b63 100644 --- a/tensorflow/contrib/lite/python/lite.py +++ b/tensorflow/contrib/lite/python/lite.py @@ -33,7 +33,7 @@ from __future__ import absolute_import from __future__ import division from __future__ import print_function -import six +from six import PY3 from google.protobuf import text_format as _text_format from google.protobuf.message import DecodeError @@ -192,7 +192,7 @@ class TocoConverter(object): print("Ignore 'tcmalloc: large alloc' warnings.") if not isinstance(file_content, str): - if six.PY3: + if PY3: file_content = file_content.decode('utf-8') else: file_content = file_content.encode('utf-8') |