minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

refactor(benchmark): prevents variable shadowing (#17607)

Browse Source
This commit is contained in:
Jacob Gordon
2026-01-22 17:00:11 -05:00
committed by GitHub
parent a921029b97
commit a296c99ff4
1 changed files with 13 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
benchmark/reasoning_benchmark/answer_extraction.py
+13 -13
View File
@@ -227,23 +227,23 @@ def extract_answer(pred_str, exhaust=False):
pred.append(program_output) pred.append(program_output)
else: # use the last number else: # use the last number
pattern = "-?\d*\.?\d+" pattern = "-?\d*\.?\d+"
ans = re.findall(pattern, pred_str.replace(",", "")) answers = re.findall(pattern, pred_str.replace(",", ""))
if len(ans) >= 1: if len(answers) >= 1:
ans = ans[-1] last_ans = answers[-1]
else: else:
ans = "" last_ans = ""
if ans: if last_ans:
pred.append(ans) pred.append(last_ans)
# multiple line # multiple line
_pred = [] _pred = []
for ans in pred: for each_ans in pred:
ans = ans.strip().split("\n")[0] each_ans = each_ans.strip().split("\n")[0]
ans = ans.lstrip(":") each_ans = each_ans.lstrip(":")
ans = ans.rstrip(".") each_ans = each_ans.rstrip(".")
ans = ans.rstrip("/") each_ans = each_ans.rstrip("/")
ans = strip_string(ans) each_ans = strip_string(each_ans)
_pred.append(ans) _pred.append(each_ans)
if exhaust: if exhaust:
return _pred return _pred
else: else: