mirror of
https://github.com/open-thought/reasoning-gym.git
synced 2026-04-25 17:10:51 +00:00
72 lines
3.1 KiB
Python
72 lines
3.1 KiB
Python
from random import Random
|
|
from typing import Dict, Any
|
|
|
|
def generate_from_variables(name: str, fish: str, day: str, w1: int, w2: int, w3: int,
|
|
n: int, unit: str, cur: str, price: float) -> Dict[str, Any]:
|
|
total = int((w1 + w2) * price + (n-2)*w3*price)
|
|
|
|
question = f"{name} caught {n} {fish}s last {day}, the first {fish} he caught weighs {w1} {unit}s, the second {fish} he caught weighs {w2} {unit}s, and the last {fish} he caught weighs {w3} {unit}s. If a {unit} of {fish} costs {cur}{price:.2f}, how much will he earn after selling all the {fish}s to the market?"
|
|
|
|
answer_cot = f"{name} will earn {w1} x {cur}{price:.2f} = {cur}{w1*price:.2f} from the first {fish}.\n" \
|
|
f"He will earn {w2} x {cur}{price:.2f} = {cur}{w2*price:.2f} for the second {fish}.\n" \
|
|
f"The rest of the {fish}s are {n}-2 = {n-2}. He will earn {w3} x {cur}{price:.2f} = {cur}{w3*price:.2f} per each of them. So he will earn {n-2} * {w3*price:.2f} = {(n-2)*w3*price:.2f}\n" \
|
|
f"Therefore, the total amount he will earn for all the {fish}s is {cur}{w1*price:.2f} + {cur}{w2*price:.2f} + {cur}{(n-2)*w3*price:.2f}= {cur}{total}.\n#### {total}"
|
|
|
|
return {
|
|
'question': question,
|
|
'answer': str(total),
|
|
'answer_cot': answer_cot,
|
|
'answer_value': total,
|
|
'variables': {
|
|
'name': name,
|
|
'fish': fish,
|
|
'day': day,
|
|
'weight1': w1,
|
|
'weight2': w2,
|
|
'weight3': w3,
|
|
'num_fish': n,
|
|
'unit': unit,
|
|
'currency': cur,
|
|
'price': price
|
|
}
|
|
}
|
|
|
|
def generate_example(rng: Random, difficulty: float = 1.0) -> Dict[str, Any]:
|
|
names = ["John", "Michael", "David", "James", "Robert", "William", "Richard"]
|
|
fish = ["salmon", "cod", "trout", "steelhead"]
|
|
days = ["Monday", "Tuesday", "Wednesday", "Thursday", "Friday"]
|
|
units = ["kilogram", "pound", "kg"]
|
|
currencies = ["$", "€", "£"]
|
|
|
|
name = rng.choice(names)
|
|
fish_type = rng.choice(fish)
|
|
day = rng.choice(days)
|
|
unit = rng.choice(units)
|
|
cur = rng.choice(currencies)
|
|
|
|
w1 = int(rng.randint(40, int(80 * difficulty)))
|
|
w2 = int(rng.randint(30, int(60 * difficulty)))
|
|
w3 = int(rng.randint(20, int(40 * difficulty)))
|
|
n = int(rng.randint(3, int(8 * difficulty)))
|
|
price = round(rng.uniform(0.25, 2.5), 2)
|
|
|
|
# Ensure result is integer
|
|
while not ((w1 + w2) * price + (n-2) * w3 * price).is_integer():
|
|
price = round(rng.uniform(0.25, 2.5), 2)
|
|
|
|
result = generate_from_variables(name, fish_type, day, w1, w2, w3, n, unit, cur, price)
|
|
|
|
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("Deandre", "tuna", "Monday", 56, 46, 26, 3,
|
|
"kilogram", "$", 0.50)
|