lazy import tensorflow

Author: Haohuan Wang

clu/metric_writers/tf/summary_writer.py

@@ -26,12 +26,12 @@
 from clu.internal import utils
 from clu.metric_writers import interface
 from etils import epy
-import tensorflow as tf
 
 with epy.lazy_imports():
   # pylint: disable=g-import-not-at-top
   from tensorboard.plugins.hparams import api as hparams_api
   from tensorboard.plugins.mesh import summary as mesh_summary  # pylint: disable=line-too-long
+  import tensorflow as tf
   # pylint: enable=g-import-not-at-top