Fix docs

Lightning-AI · Sep 24, 2024 · c7bdce9 · c7bdce9
1 parent a07e021
commit c7bdce9
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/docs/source/segmentation/hausdorff_distance.rst b/docs/source/segmentation/hausdorff_distance.rst
@@ -1,9 +1,7 @@
 .. customcarditem::
    :header: Hausdorff Distance
    :image: https://pl-flash-data.s3.amazonaws.com/assets/thumbnails/text_classification.svg
-   :tags: Retrieval
-
-.. include:: ../links.rst
+   :tags: segmentation
 
 ##################
 Hausdorff Distance

diff --git a/src/torchmetrics/segmentation/hausdorff_distance.py b/src/torchmetrics/segmentation/hausdorff_distance.py
@@ -47,6 +47,7 @@ class HausdorffDistance(Metric):
 
     Example:
         >>> import torch
+        >>> from torchmetrics.segmentation import HausdorffDistance
         >>> preds = torch.tensor([[1, 1, 1, 1, 1],
         ...                       [1, 0, 0, 0, 1],
         ...                       [1, 0, 0, 0, 1],