open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-19 12:58:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

adjusted geom prompt

Browse source
This commit is contained in:
joesharratt1229 2025-02-19 19:52:18 +00:00
parent ab44ec6fdd
commit a4419d6e4b
1 changed files with 2 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

3
reasoning_gym/geometry/advanced_geometry.py
View file

@ -44,6 +44,7 @@ GEOMETRY_FORMAT_INSTRUCTIONS = "\n".join(
"1. Give coordinates in the form (x, y)",
"2. Round decimal answers to 3 decimal places",
"3. Use the degree symbol ° for angles",
"4. Return only th angle, coordinates, or radius as your answer.",
]
)
@ -264,7 +265,7 @@ class AdvancedGeometryDataset(ProceduralDataset):
elif metadata["task_type"] == "incircle_radius":
if answer == expected_answer:
reward = 1.0
elif np.abs(float(answer) - float(metadata["incircle_radius_exact"])) < EPSILON:
elif np.round(float(answer), 2) == np.round(float(metadata["incircle_radius_exact"]), 2):
reward = 0.5
elif len(answer.strip()) > 0:
reward = 0.05