typo

Lightning-AI · Mar 17, 2022 · d3e8d37 · d3e8d37
1 parent 116c88f
commit d3e8d37
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchmetrics/functional/classification/kl_divergence.py b/torchmetrics/functional/classification/kl_divergence.py
@@ -48,7 +48,7 @@ def _kld_update(p: Tensor, q: Tensor, log_prob: bool) -> Tuple[Tensor, int]:
     return measures, total
 
 
-def _kld_compute(measures: Tensor, total: Tensor, reduction: Literal["_mean", "sum", "none", None] = "mean") -> Tensor:
+def _kld_compute(measures: Tensor, total: Tensor, reduction: Literal["mean", "sum", "none", None] = "mean") -> Tensor:
     """Computes the KL divergenece based on the type of reduction.
 
     Args: