From 3720ba92bbf046b2eb105603d7f70ffa4b590a9f Mon Sep 17 00:00:00 2001 From: Jirka Borovec <6035284+Borda@users.noreply.github.com> Date: Thu, 10 Oct 2024 19:20:30 +0200 Subject: [PATCH] typo --- src/torchmetrics/functional/text/helper_embedding_metric.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/torchmetrics/functional/text/helper_embedding_metric.py b/src/torchmetrics/functional/text/helper_embedding_metric.py index 7ae82305769..f2b59126c7d 100644 --- a/src/torchmetrics/functional/text/helper_embedding_metric.py +++ b/src/torchmetrics/functional/text/helper_embedding_metric.py @@ -195,7 +195,7 @@ def __init__( tokenizer: Any, max_length: int = 512, preprocess_text_fn: Callable[ - [List[str], Any, int.bool], Union[Dict[str, Tensor], Tuple[Dict[str, Tensor], Optional[Tensor]]] + [List[str], Any, int, bool], Union[Dict[str, Tensor], Tuple[Dict[str, Tensor], Optional[Tensor]]] ] = _preprocess_text, idf: bool = False, tokens_idf: Optional[Dict[int, float]] = None,