From 943a1757d523bfe46d54ee4f51e67d948ee06bb6 Mon Sep 17 00:00:00 2001
From: Yifei Feng <yifeif@google.com>
Date: Wed, 10 Jul 2019 15:37:15 -0700
Subject: [PATCH] Fix import error for tf.summary.create_file_writer for 2.0.

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

PiperOrigin-RevId: 257496023
---
 tensorflow/api_template.__init__.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/tensorflow/api_template.__init__.py b/tensorflow/api_template.__init__.py
index c6d39cd9f48..3d8d92c63e7 100644
--- a/tensorflow/api_template.__init__.py
+++ b/tensorflow/api_template.__init__.py
@@ -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