fix doctest import

Lightning-AI · Mar 17, 2023 · 9a72a60 · 9a72a60
1 parent 6209059
commit 9a72a60
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/torchmetrics/classification/recall_at_fixed_precision.py b/src/torchmetrics/classification/recall_at_fixed_precision.py
@@ -292,7 +292,7 @@ def plot(
         .. plot::
             :scale: 75
 
-            >>> from torch import randint
+            >>> from torch import rand, randint
             >>> # Example plotting a multiple values per class
             >>> from torchmetrics.classification import MulticlassRecallAtFixedPrecision
             >>> metric = MulticlassRecallAtFixedPrecision(num_classes=3, min_precision=0.5)