diff --git a/clu/metric_writers/tf/summary_writer.py b/clu/metric_writers/tf/summary_writer.py index 5711e70..081e8e9 100644 --- a/clu/metric_writers/tf/summary_writer.py +++ b/clu/metric_writers/tf/summary_writer.py @@ -25,9 +25,13 @@ from clu.internal import utils from clu.metric_writers import interface +from etils import epy import tensorflow as tf -from tensorboard.plugins.hparams import api as hparams_api +with epy.lazy_imports(): + # pylint: disable=g-import-not-at-top + from tensorboard.plugins.hparams import api as hparams_api + # pylint: enable=g-import-not-at-top Array = interface.Array