modify doctest

Lightning-AI · Jan 10, 2025 · 1156b18 · 1156b18
1 parent 1571f8c
commit 1156b18
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 3 deletions.
diff --git a/src/torchmetrics/functional/multimodal/clip_score.py b/src/torchmetrics/functional/multimodal/clip_score.py
@@ -266,7 +266,6 @@ def clip_score(
         tensor(24.4255)
 
     Example:
-        >>> import torch
         >>> from torchmetrics.functional.multimodal import clip_score
         >>> image1 = torch.randint(255, (3, 224, 224), generator=torch.Generator().manual_seed(42))
         >>> image2 = torch.randint(255, (3, 224, 224), generator=torch.Generator().manual_seed(43))

diff --git a/src/torchmetrics/multimodal/clip_score.py b/src/torchmetrics/multimodal/clip_score.py
@@ -101,7 +101,6 @@ class CLIPScore(Metric):
             If transformers package is not installed or version is lower than 4.10.0
 
     Example:
-        >>> from torch import randint
         >>> from torchmetrics.multimodal.clip_score import CLIPScore
         >>> metric = CLIPScore(model_name_or_path="openai/clip-vit-base-patch16")
         >>> image = torch.randint(255, (3, 224, 224), generator=torch.Generator().manual_seed(42))
@@ -110,7 +109,6 @@ class CLIPScore(Metric):
         tensor(24.)
 
     Example:
-        >>> import torch
         >>> from torchmetrics.multimodal.clip_score import CLIPScore
         >>> metric = CLIPScore(model_name_or_path="openai/clip-vit-base-patch16")
         >>> image1 = torch.randint(255, (3, 224, 224), generator=torch.Generator().manual_seed(42))