experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix import error for tf.summary.create_file_writer for 2.0.

Browse Source 
tensorboard/summary/_tf/summary/__init__.py's reexport_tf_summary depends on tensorflow.compat.v2 being already imported (in sys.modules).

PiperOrigin-RevId: 257496023
This commit is contained in:
Yifei Feng 2019-07-10 15:37:15 -07:00 committed by TensorFlower Gardener
parent d697392fca
commit 943a1757d5
1 changed files with 4 additions and 2 deletions

6
tensorflow/api_template.__init__.py
View File

@ -56,6 +56,10 @@ elif _tf_api_dir not in __path__:
__path__.append(_tf_api_dir)
# Hook external TensorFlow modules.
# Import compat before trying to import summary from tensorboard, so that
# reexport_tf_summary can get compat from sys.modules
_current_module.compat.v2.compat.v1 = _current_module.compat.v1
try:
from tensorboard.summary._tf import summary
_current_module.__path__ = (
@ -150,6 +154,4 @@ if hasattr(_current_module, 'keras'):
setattr(_current_module, "metrics", metrics)
setattr(_current_module, "optimizers", optimizers)
setattr(_current_module, "initializers", initializers)
_current_module.compat.v2.compat.v1 = _current_module.compat.v1
# pylint: enable=undefined-variable