open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-25 17:10:51 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Create python generator files for gsm symbolic templates

Browse source
This commit is contained in:
abdulhakeem 2025-02-01 20:23:30 -06:00
parent ff13dc6825
commit 1e0dbc9875
100 changed files with 6919 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

85
reasoning_gym/arithmetic/gsm_symbolic/generator_37.py Normal file
View file

@ -0,0 +1,85 @@
from random import Random
from typing import Dict, Any
def generate_from_variables(worker: str, base: int, unit: str, tool1: str,
tool2: str, tool3: str, mult1: int, mult2: int,
n: int, days: int) -> Dict[str, Any]:
iron_amount = base * mult1
steel_amount = int(iron_amount * (1 + mult2/100))
daily_total = steel_amount * n
total_amount = daily_total * days
question = f"One {worker} can mine {base} {unit} of ore per day with {tool1}. He can mine {mult1} times as much with a {tool2} and {mult2}% more with a {tool3} than with a {tool2}. How many {unit} of ore can {n} {worker}s with {tool3}s mine in a month with {days} days?"
answer_cot = f"First find how much ore a {worker} can mine with a {tool2}: {base} {unit}/day * {mult1} = {iron_amount} {unit}/day\n" \
f"Then multiply that amount by {100+mult2}% to find how much a {worker} can mine with a {tool3}: {iron_amount} {unit}/day * {100+mult2}% = {steel_amount} {unit}/day\n" \
f"Then multiply the amount one {worker} can mine in a day with a {tool3} by the number of {worker}s: {steel_amount} {unit}/day/{worker} * {n} {worker}s = {daily_total} {unit}/day\n" \
f"Then multiply the daily amount of ore by the number of days to find the total ore mined in a month: {daily_total} {unit}/day * {days} days = {total_amount} {unit}/day\n" \
f"#### {total_amount}"
return {
'question': question,
'answer': str(total_amount),
'answer_cot': answer_cot,
'answer_value': total_amount,
'variables': {
'worker': worker,
'base_amount': base,
'unit': unit,
'tool1': tool1,
'tool2': tool2,
'tool3': tool3,
'mult1': mult1,
'mult2': mult2,
'num_workers': n,
'num_days': days,
'iron_amount': iron_amount,
'steel_amount': steel_amount,
'daily_total': daily_total
}
}
def generate_example(rng: Random, difficulty: float = 1.0) -> Dict[str, Any]:
workers = ["miner", "goblin", "gnome", "troll"]
tools1 = ["bare hands", "basic shovel", "wooden pickaxe"]
units = ["pounds", "kgs"]
tools2 = ["nickel pickaxe", "bronze pickaxe", "silver pickaxe"]
tools3 = ["steel pickaxe", "diamond pickaxe", "mithril pickaxe", "titanium pickaxe"]
worker = rng.choice(workers)
tool1 = rng.choice(tools1)
unit = rng.choice(units)
tool2 = rng.choice(tools2)
tool3 = rng.choice(tools3)
base = int(rng.randint(5, int(20 * difficulty)))
mult1 = int(rng.randint(2, int(4 * difficulty)))
mult2 = int(rng.randint(30, int(80 * difficulty)) // 5 * 5)
n = int(rng.randint(20, int(50 * difficulty)))
days = int(rng.randint(28, 32))
# Verify conditions
while not (base * mult1 * (1 + mult2/100)).is_integer() or \
int(base * mult1 * (1 + mult2/100) * n * days) >= 100000:
base = int(rng.randint(5, int(20 * difficulty)))
mult1 = int(rng.randint(2, int(4 * difficulty)))
mult2 = int(rng.randint(30, int(80 * difficulty)) // 5 * 5)
n = int(rng.randint(20, int(50 * difficulty)))
result = generate_from_variables(worker, base, unit, tool1, tool2, tool3,
mult1, mult2, n, days)
return {
'question': result['question'],
'answer': result['answer'],
'metadata': {
'difficulty': difficulty,
'answer_value': result['answer_value'],
'answer_cot': result['answer_cot'],
'variables': result['variables']
}
}
def original_example() -> Dict[str, Any]:
return generate_from_variables("dwarf", 12, "pounds", "his bare hands",
"iron pickaxe", "steel pickaxe", 2, 50, 40, 30)