lazy import tensorflow

google · Nov 22, 2024 · e408416 · e408416
1 parent 307b0bc
commit e408416
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/clu/metric_writers/tf/summary_writer.py b/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