Skip to content

helpers

ipw.evaluation.helpers

normalize_math_answer(text)

Best-effort normalization for simple mathematical expressions.

Source code in intelligence-per-watt/src/ipw/evaluation/helpers.py 
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
def normalize_math_answer(text: Optional[str]) -> Optional[str]:
    """Best-effort normalization for simple mathematical expressions."""

    if text is None:
        return None
    if not isinstance(text, str):
        text = str(text)
    s = text.strip()
    if not s:
        return None

    # Remove LaTeX math delimiters
    s = s.replace("$", "")

    # Extract content inside the last \boxed{...} if present
    boxed_matches = list(re.finditer(r"\\boxed\{([^}]*)\}", s))
    if boxed_matches:
        s = boxed_matches[-1].group(1)

    # Remove thin spaces / !, etc.
    s = re.sub(r"\\[!,]", "", s)

    # Replace \frac{a}{b} with a/b for simple integer fractions
    s = re.sub(r"\\frac\{(\d+)\}\{(\d+)\}", r"\1/\2", s)

    # Replace \sqrt{n} with √n for small integers
    s = re.sub(r"\\sqrt\{(\d+)\}", r"√\1", s)

    # Remove thousands separators
    s = s.replace(",", "")

    # Collapse whitespace
    s = re.sub(r"\s+", " ", s).strip()
    return s or None