diff --git a/simlm/src/metrics.py b/simlm/src/metrics.py index ab9a4fc0d..d355c4fab 100644 --- a/simlm/src/metrics.py +++ b/simlm/src/metrics.py @@ -9,7 +9,7 @@ def trec_eval(qrels: Dict[str, Dict[str, int]], predictions: Dict[str, List[ScoredDoc]], - k_values: Tuple[int] = (10, 50, 100, 200, 1000)) -> Dict[str, float]: + k_values: Tuple[int, ...] = (10, 50, 100, 200, 1000)) -> Dict[str, float]: ndcg, _map, recall = {}, {}, {} for k in k_values: