skip doctests on missing package

src/torchmetrics/clustering/cluster_accuracy.py

@@ -19,12 +19,15 @@
 from torchmetrics.functional.classification import multiclass_confusion_matrix
 from torchmetrics.functional.clustering.cluster_accuracy import _cluster_accuracy_compute
 from torchmetrics.metric import Metric
-from torchmetrics.utilities.imports import _MATPLOTLIB_AVAILABLE
+from torchmetrics.utilities.imports import _MATPLOTLIB_AVAILABLE, _TORCH_LINEAR_ASSIGNMENT_AVAILABLE
 from torchmetrics.utilities.plot import _AX_TYPE, _PLOT_OUT_TYPE
 
 if not _MATPLOTLIB_AVAILABLE:
     __doctest_skip__ = ["ClusterAccuracy.plot"]
 
+if not _TORCH_LINEAR_ASSIGNMENT_AVAILABLE:
+    __doctest_skip__ = ["ClusterAccuracy"]
+
 
 class ClusterAccuracy(Metric):
     r"""Compute `Cluster Accuracy`_ between predicted and target clusters.

src/torchmetrics/functional/clustering/cluster_accuracy.py

@@ -17,6 +17,10 @@
 
 from torchmetrics.functional.classification import multiclass_confusion_matrix
 from torchmetrics.functional.clustering.utils import check_cluster_labels
+from torchmetrics.utilities.imports import _TORCH_LINEAR_ASSIGNMENT_AVAILABLE
+
+if not _TORCH_LINEAR_ASSIGNMENT_AVAILABLE:
+    __doctest_skip__ = ["cluster_accuracy"]
 
 
 def _cluster_accuracy_compute(confmat: Tensor) -> Tensor:

src/torchmetrics/utilities/imports.py

@@ -66,5 +66,6 @@
 _SENTENCEPIECE_AVAILABLE = RequirementCache("sentencepiece")
 _SCIPI_AVAILABLE = RequirementCache("scipy")
 _SKLEARN_GREATER_EQUAL_1_3 = RequirementCache("scikit-learn>=1.3.0")
+_TORCH_LINEAR_ASSIGNMENT_AVAILABLE = RequirementCache("torch_linear_assignment")
 
 _LATEX_AVAILABLE: bool = shutil.which("latex") is not None