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,