From 5b63e9fa7245a15ecc70d59776d7b905ef549997 Mon Sep 17 00:00:00 2001 From: Marvin Ritter Date: Fri, 21 Jul 2023 09:32:42 -0700 Subject: [PATCH] Fix TorchTensorboardWriter. PiperOrigin-RevId: 549969243 --- clu/metric_writers/torch_tensorboard_writer.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/clu/metric_writers/torch_tensorboard_writer.py b/clu/metric_writers/torch_tensorboard_writer.py index 39e5126..baa8acb 100644 --- a/clu/metric_writers/torch_tensorboard_writer.py +++ b/clu/metric_writers/torch_tensorboard_writer.py @@ -65,8 +65,9 @@ def write_audios( key, value, global_step=step, sample_rate=sample_rate) def write_texts(self, step: int, texts: Mapping[str, str]): - for key, value in texts.items(): - self._writer.text(key, value, global_step=step) + raise NotImplementedError( + "TorchTensorBoardWriter does not support writing texts." + ) def write_histograms(self, step: int,