Fix type hint for play_a_match_single (#3008)

lm-sys · Feb 4, 2024 · 81785d7 · 81785d7
1 parent 3bef934
commit 81785d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fastchat/llm_judge/common.py b/fastchat/llm_judge/common.py
@@ -189,7 +189,7 @@ def run_judge_single(question, answer, judge, ref_answer, multi_turn=False):
     return rating, user_prompt, judgment
 
 
-def play_a_match_single(match: MatchPair, output_file: str):
+def play_a_match_single(match: MatchSingle, output_file: str):
     question, model, answer, judge, ref_answer, multi_turn = (
         match.question,
         match.model,