From 580eaf0c96c1443efdbd500d4f747ffb75d75178 Mon Sep 17 00:00:00 2001 From: CLU Authors Date: Tue, 19 Mar 2024 16:42:36 -0700 Subject: [PATCH] This fix adjusts import to avoid accidentally pulling in keras in clu. https://github.com/google/CommonLoopUtils/pull/336 PiperOrigin-RevId: 617332775 --- clu/metric_writers/tf/summary_writer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/clu/metric_writers/tf/summary_writer.py b/clu/metric_writers/tf/summary_writer.py index 5711e70..436e519 100644 --- a/clu/metric_writers/tf/summary_writer.py +++ b/clu/metric_writers/tf/summary_writer.py @@ -27,7 +27,7 @@ from clu.metric_writers import interface import tensorflow as tf -from tensorboard.plugins.hparams import api as hparams_api +from tensorboard.plugins.hparams import summary_v2 Array = interface.Array @@ -95,7 +95,7 @@ def write_histograms( def write_hparams(self, hparams: Mapping[str, Any]): with self._summary_writer.as_default(): - hparams_api.hparams(dict(utils.flatten_dict(hparams))) + summary_v2.hparams(dict(utils.flatten_dict(hparams))) def flush(self): self._summary_writer.flush()