From 555e24376fa7ba923c08afc1632f322c450c69e3 Mon Sep 17 00:00:00 2001
From: Roman A <121314722+GameRoMan@users.noreply.github.com>
Date: Fri, 9 May 2025 21:31:53 +0100
Subject: [PATCH] Fix type annotations

---
 simlm/src/metrics.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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: