Apply suggestions from code review

Co-authored-by: Carlos Mocholí <[email protected]>
Lightning-AI · Mar 23, 2021 · 543183f · 543183f
1 parent 7c30ef2
commit 543183f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/pytorch_lightning/metrics/regression/r2score.py b/pytorch_lightning/metrics/regression/r2score.py
@@ -32,8 +32,8 @@ def __init__(
         dist_sync_fn: Callable = None,
     ):
         """
-        This implementation refers to :class:`~torchmetrics.Accuracy`.
+        This implementation refers to :class:`~torchmetrics.R2Score`.
 
         .. deprecated::
-            Use :class:`~torchmetrics.Accuracy`. Will be removed in v1.5.0.
+            Use :class:`~torchmetrics.R2Score`. Will be removed in v1.5.0.
         """
diff --git a/pytorch_lightning/metrics/regression/ssim.py b/pytorch_lightning/metrics/regression/ssim.py
@@ -34,8 +34,8 @@ def __init__(
         process_group: Optional[Any] = None,
     ):
         """
-        This implementation refers to :class:`~torchmetrics.Accuracy`.
+        This implementation refers to :class:`~torchmetrics.SSIM`.
 
         .. deprecated::
-            Use :class:`~torchmetrics.Accuracy`. Will be removed in v1.5.0.
+            Use :class:`~torchmetrics.SSIM`. Will be removed in v1.5.0.
         """