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 native types List->list, Dict->dict, Set->set, Tuple->tuple

Browse source
This commit is contained in:
Andreas Koepf 2025-02-21 15:13:19 +01:00
parent 6b5c7a8637
commit ff5b210106
95 changed files with 754 additions and 760 deletions
Download patch file Download diff file Expand all files Collapse all files

4
reasoning_gym/arithmetic/basic_arithmetic.py
View file

@ -1,6 +1,6 @@
from dataclasses import dataclass
from random import Random
from typing import Any, Dict, Literal, Optional
from typing import Any, Literal, Optional
from reasoning_gym import utils
@ -234,7 +234,7 @@ class BasicArithmeticDataset(ProceduralDataset):
template = rng.choice(templates)
return template.format(expression)
def score_answer(self, answer: Optional[str], entry: Dict[str, any]) -> float:
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)