NousResearch/atropos
1
0
Fork 0
mirror of https://github.com/NousResearch/atropos.git synced 2026-04-19 12:57:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Merge branch 'port_many_evals' of https://github.com/NousResearch/atropos into port_many_evals

Browse source
This commit is contained in:
teknium 2025-12-28 04:11:32 +00:00
commit ea6db6fe92
2 changed files with 45 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

8
environments/eval_environments/eed_score.py
View file

@ -691,7 +691,9 @@ def master_convert(s: str):
Various exceptions if conversion fails
"""
if not EED_AVAILABLE:
raise ImportError("latex2sympy2_extended and sympy are required for EED scoring")
raise ImportError(
"latex2sympy2_extended and sympy are required for EED scoring"
)
preprocessed_stage1 = first_preprocess(s)
preprocessed_stage2 = second_preprocess(preprocessed_stage1)
@ -723,7 +725,9 @@ def sympy_to_tree(expr) -> TreeNode:
ValueError: If expression contains unsupported types
"""
# Numbers and constants
if isinstance(expr, (Integer, Pi, Exp1, Float, Rational, Infinity, NegativeInfinity)):
if isinstance(
expr, (Integer, Pi, Exp1, Float, Rational, Infinity, NegativeInfinity)
):
return TreeNode(label=f"number_{expr}", children=[])
# Symbols