From cb71bc3beb665ff912a0cac1f0d5f3fb0fec433c Mon Sep 17 00:00:00 2001 From: Etienne Pot Date: Fri, 12 Apr 2024 07:25:06 -0700 Subject: [PATCH] Rename differentially_private_aggregate -> optax.contrib PiperOrigin-RevId: 624171042 --- 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