From 6d60573516e7639232b178beeba50c7fd11d4a9f Mon Sep 17 00:00:00 2001 From: Etienne Pot Date: Mon, 15 Apr 2024 02:10:43 -0700 Subject: [PATCH] Make g3pdb lazy again PiperOrigin-RevId: 624881107 --- clu/metric_writers/tf/summary_writer.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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