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

use Decimal class for numeric comparison e.g. +0123.100 == 123.1

Browse source
This commit is contained in:
Andreas Koepf 2025-02-21 15:33:42 +01:00
parent 3e7ff3b084
commit acde58a200
4 changed files with 15 additions and 18 deletions
Download patch file Download diff file Expand all files Collapse all files

6
reasoning_gym/arithmetic/basic_arithmetic.py
View file

@ -2,8 +2,6 @@ from dataclasses import dataclass
from random import Random
from typing import Any, Literal, Optional
from reasoning_gym import utils
from ..factory import ProceduralDataset, register_dataset
@ -234,10 +232,6 @@ class BasicArithmeticDataset(ProceduralDataset):
template = rng.choice(templates)
return template.format(expression)
def score_answer(self, answer: Optional[str], entry: dict[str, Any]) -> float:
oracle_answer = entry["answer"].strip()
return utils.compute_reward(answer, oracle_answer, allow_commas=False)
# Register the dataset
register_dataset("basic_arithmetic", BasicArithmeticDataset, BasicArithmeticDatasetConfig)