def is_correct()

in grade_school_math/dataset.py [0:0]

• 4 lines of code
• 1 McCabe index (conditional complexity)

def is_correct(model_completion, gt_example):
    gt_answer = extract_answer(gt_example["answer"])
    assert gt_answer != INVALID_ANS
    return extract_answer(model_completion) == gt_answer