fix doctests

Lightning-AI · Oct 12, 2024 · 2e6e5e3 · 2e6e5e3
1 parent 6043801
commit 2e6e5e3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/torchmetrics/regression/nrmse.py b/src/torchmetrics/regression/nrmse.py
@@ -120,8 +120,8 @@ class NormalizedRootMeanSquaredError(Metric):
 
         >>> import torch
         >>> from torchmetrics import NormalizedRootMeanSquaredError
-        >>> target = tensor([2.5, 5.0, 4.0, 8.0])
-        >>> preds = tensor([3.0, 5.0, 2.5, 7.0])
+        >>> target = torch.tensor([2.5, 5.0, 4.0, 8.0])
+        >>> preds = torch.tensor([3.0, 5.0, 2.5, 7.0])
         >>> nrmse = NormalizedRootMeanSquaredError(normalization="mean")
         >>> nrmse(preds, target)
         tensor(0.1919)