diff --git a/clu/metric_writers/tf/summary_writer.py b/clu/metric_writers/tf/summary_writer.py index 7e00570..c767d33 100644 --- 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