diff --git a/docs/source/segmentation/hausdorff_distance.rst b/docs/source/segmentation/hausdorff_distance.rst index 182e547c4b3..ba1eb35a6b4 100644 --- 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 index 55987f74bdc..b7542845a0a 100644 --- 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],