imports

Lightning-AI · Mar 13, 2023 · 925522f · 925522f
1 parent 0f6e965
commit 925522f
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 8 deletions.
diff --git a/src/torchmetrics/retrieval/precision.py b/src/torchmetrics/retrieval/precision.py
@@ -129,18 +129,19 @@ def plot(
         .. plot::
             :scale: 75
 
-            >>> # Example plotting a single value
             >>> import torch
             >>> from torchmetrics.retrieval import RetrievalPrecision
+            >>> # Example plotting a single value
             >>> metric = RetrievalPrecision()
             >>> metric.update(torch.rand(10,), torch.randint(2, (10,)), indexes=torch.randint(2,(10,)))
             >>> fig_, ax_ = metric.plot()
 
         .. plot::
             :scale: 75
 
-            >>> # Example plotting multiple values
+            >>> import torch
             >>> from torchmetrics.retrieval import RetrievalPrecision
+            >>> # Example plotting multiple values
             >>> metric = RetrievalPrecision()
             >>> values = []
             >>> for _ in range(10):

diff --git a/src/torchmetrics/retrieval/r_precision.py b/src/torchmetrics/retrieval/r_precision.py
@@ -102,18 +102,19 @@ def plot(
         .. plot::
             :scale: 75
 
-            >>> # Example plotting a single value
             >>> import torch
             >>> from torchmetrics.retrieval import RetrievalRPrecision
+            >>> # Example plotting a single value
             >>> metric = RetrievalRPrecision()
             >>> metric.update(torch.rand(10,), torch.randint(2, (10,)), indexes=torch.randint(2,(10,)))
             >>> fig_, ax_ = metric.plot()
 
         .. plot::
             :scale: 75
 
-            >>> # Example plotting multiple values
+            >>> import torch
             >>> from torchmetrics.retrieval import RetrievalRPrecision
+            >>> # Example plotting multiple values
             >>> metric = RetrievalRPrecision()
             >>> values = []
             >>> for _ in range(10):

diff --git a/src/torchmetrics/retrieval/recall.py b/src/torchmetrics/retrieval/recall.py
@@ -121,18 +121,19 @@ def plot(
         .. plot::
             :scale: 75
 
-            >>> # Example plotting a single value
             >>> import torch
             >>> from torchmetrics.retrieval import RetrievalRecall
+            >>> # Example plotting a single value
             >>> metric = RetrievalRecall()
             >>> metric.update(torch.rand(10,), torch.randint(2, (10,)), indexes=torch.randint(2,(10,)))
             >>> fig_, ax_ = metric.plot()
 
         .. plot::
             :scale: 75
 
-            >>> # Example plotting multiple values
+            >>> import torch
             >>> from torchmetrics.retrieval import RetrievalRecall
+            >>> # Example plotting multiple values
             >>> metric = RetrievalRecall()
             >>> values = []
             >>> for _ in range(10):

diff --git a/src/torchmetrics/retrieval/reciprocal_rank.py b/src/torchmetrics/retrieval/reciprocal_rank.py
@@ -101,18 +101,19 @@ def plot(
         .. plot::
             :scale: 75
 
-            >>> # Example plotting a single value
             >>> import torch
             >>> from torchmetrics.retrieval import RetrievalMRR
+            >>> # Example plotting a single value
             >>> metric = RetrievalMRR()
             >>> metric.update(torch.rand(10,), torch.randint(2, (10,)), indexes=torch.randint(2,(10,)))
             >>> fig_, ax_ = metric.plot()
 
         .. plot::
             :scale: 75
 
-            >>> # Example plotting multiple values
+            >>> import torch
             >>> from torchmetrics.retrieval import RetrievalMRR
+            >>> # Example plotting multiple values
             >>> metric = RetrievalMRR()
             >>> values = []
             >>> for _ in range(10):