correct typing

huggingface · Jan 27, 2025 · 93d108e · 93d108e
1 parent 0a49ac0
commit 93d108e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/lighteval/metrics/metrics_sample.py b/src/lighteval/metrics/metrics_sample.py
@@ -26,7 +26,7 @@
 
 import logging
 import os
-from typing import Callable, Literal
+from typing import Callable, Literal, Union
 
 import nltk
 import numpy as np
@@ -1050,10 +1050,10 @@ def __init__(
         self,
         k: int,
         n: int = None,
-        normalize_gold: callable = None,
-        normalize_pred: callable = None,
+        normalize_gold: Callable = None,
+        normalize_pred: Callable = None,
         strip_strings: bool = False,
-        sample_scoring_function: callable | str = None,
+        sample_scoring_function: Union[Callable[[str, str], float], str] = None,
     ):
         """Computing pass at k