From 7b5f8ebbb3065e5df6c6bfa39e2f725319ae8f2e Mon Sep 17 00:00:00 2001 From: Felix Divo <4403130+felixdivo@users.noreply.github.com> Date: Mon, 16 Jan 2023 18:36:20 +0100 Subject: [PATCH] Fix doc formatting (#1438) Co-authored-by: Jirka Borovec <6035284+Borda@users.noreply.github.com> --- src/torchmetrics/metric.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/torchmetrics/metric.py b/src/torchmetrics/metric.py index 0a3d13d1e38..17d1e9eac19 100644 --- a/src/torchmetrics/metric.py +++ b/src/torchmetrics/metric.py @@ -51,8 +51,8 @@ class Metric(Module, ABC): call of ``update()`` and are synchronized across processes when ``compute()`` is called. Note: - Metric state variables can either be ``torch.Tensors`` or an empty list which can we used - to store `torch.Tensors``. + Metric state variables can either be :class:`~torch.Tensor` or an empty list which can we used + to store :class:`~torch.Tensor`. Note: Different metrics only override ``update()`` and not ``forward()``. A call to ``update()``