add missing import

Lightning-AI · Oct 9, 2024 · ebb4f4c · ebb4f4c
1 parent 6f188a8
commit ebb4f4c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/torchmetrics/utilities/distributed.py b/src/torchmetrics/utilities/distributed.py
@@ -18,6 +18,8 @@
 from torch.nn import functional as F  # noqa: N812
 from typing_extensions import Literal
 
+from torchmetrics.utilities.imports import _TORCH_GREATER_EQUAL_2_1
+
 
 def reduce(x: Tensor, reduction: Literal["elementwise_mean", "sum", "none", None]) -> Tensor:
     """Reduces a given tensor by a given reduction method.