@k is added

map.py

@@ -95,7 +95,8 @@ class map(evaluate.Metric):
         
         run = Run(preds)
         qrels = Qrels(refs)
-        map_score = ran_evaluate(qrels, run, "map")
+        metric = "map" if k is None else f"map@{k}"
+        map_score = ran_evaluate(qrels, run, metric)
         return {
             "map": map_score,
         }