Function parse_math_answer

TransMLA_NeurIPS_2025/lighteval/math_utils.py:420–435

(text: str, data_name)

def parse_math_answer(text: str, data_name):
    # parse ground truth
    if data_name in ["math", "minerva_math"]:
        gt_cot = text
        gt_ans = extract_answer(gt_cot, data_name)
    elif data_name == "gsm8k":
        gt_cot, gt_ans = text.split("####")
    else:
        raise NotImplementedError(f"`{data_name}`")
    # post process
    gt_ans = (
        gt_ans.replace("\\neq", "\\ne")
        .replace("\\leq", "\\le")
        .replace("\\geq", "\\ge")
    )
    return gt_ans