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 5d02064b5a
commit 3e7ff3b084
95 changed files with 754 additions and 760 deletions
Download patch file Download diff file Expand all files Collapse all files

6
reasoning_gym/arithmetic/fraction_simplification.py
View file

@ -4,7 +4,7 @@ import re
from dataclasses import dataclass
from math import gcd
from random import Random
from typing import Any, Dict, Optional, Sequence, Tuple
from typing import Any, Optional, Sequence
from ..factory import ProceduralDataset, register_dataset
@ -42,7 +42,7 @@ class FractionSimplificationDataset(ProceduralDataset):
def __init__(self, config: FractionSimplificationConfig):
super().__init__(config=config, seed=config.seed, size=config.size)
def _generate_fraction(self, rng: Random) -> Tuple[int, int, int, int]:
def _generate_fraction(self, rng: Random) -> tuple[int, int, int, int]:
"""Generate a random fraction and its simplified form.
Returns (numerator, denominator, simplified_num, simplified_den)"""
# Try to generate valid fractions until we get one that meets our criteria
@ -134,7 +134,7 @@ class FractionSimplificationDataset(ProceduralDataset):
except:
return None
def score_answer(self, answer: Optional[str], entry: Dict[str, Any]):
def score_answer(self, answer: Optional[str], entry: dict[str, Any]):
reward = 0.0
metadata = entry["metadata"]
try: