mirror of
https://github.com/open-thought/reasoning-gym.git
synced 2026-04-25 17:10:51 +00:00
74 lines
3.4 KiB
Python
74 lines
3.4 KiB
Python
from random import Random
|
|
from typing import Dict, Any
|
|
|
|
def generate_from_variables(name: str, item: str, price: float, percent: float,
|
|
usage: int, extra_item: str, extra_price: float,
|
|
currency: str, unit: str) -> Dict[str, Any]:
|
|
price_increase = price * percent / 100
|
|
new_price = price + price_increase
|
|
weekly_usage = usage * 7
|
|
coffee_cost = new_price * weekly_usage
|
|
total_cost = coffee_cost + extra_price
|
|
|
|
question = f"{name} goes to the store to buy some {item}. The normal brand of {item} he buys costs {currency}{price} per {unit}. He had to buy a more expensive brand that costs {int(percent)}% more since his favorite brand was sold out. He decides to buy a week's worth of {item} and he uses {usage} {unit} of {item} per day. He also decided to buy himself a {extra_item} for {currency}{extra_price}. How much did everything cost?"
|
|
|
|
answer_cot = f"The {item} he bought was {price}*{percent/100}={price_increase} more expensive per {unit} than what he normally buys\nSo it cost {price}+{price_increase}={new_price} per {unit}\nHe goes through {usage}*7={weekly_usage} {unit}s of {item} a week\nSo he paid {new_price}*{weekly_usage}={coffee_cost} on {item}\nThat means his total bill was {coffee_cost}+{extra_price}={total_cost}\n#### {int(total_cost)}"
|
|
|
|
return {
|
|
'question': question,
|
|
'answer': str(int(total_cost)),
|
|
'answer_cot': answer_cot,
|
|
'answer_value': int(total_cost),
|
|
'variables': {
|
|
'name': name,
|
|
'item': item,
|
|
'base_price': price,
|
|
'percent_increase': percent,
|
|
'usage_per_day': usage,
|
|
'extra_item': extra_item,
|
|
'extra_price': extra_price,
|
|
'currency': currency,
|
|
'unit': unit,
|
|
'weekly_usage': weekly_usage,
|
|
'total_cost': total_cost
|
|
}
|
|
}
|
|
|
|
def generate_example(rng: Random, difficulty: float = 1.0) -> Dict[str, Any]:
|
|
names_male = ['John', 'Michael', 'David', 'James', 'Robert', 'William', 'Richard', 'Thomas']
|
|
items = ['tea', 'sugar', 'flour', 'rice']
|
|
currencies_sym = ['$', '£', '€']
|
|
units = ['kilogram', 'kg']
|
|
extra_items = ['cookie', 'muffin', 'bagel']
|
|
|
|
name = rng.choice(names_male)
|
|
item = rng.choice(items)
|
|
currency = rng.choice(currencies_sym)
|
|
unit = rng.choice(units)
|
|
extra_item = rng.choice(extra_items)
|
|
|
|
price = int(rng.randint(3, int(25 * difficulty)))
|
|
percent = int(rng.randint(2, int(10 * difficulty))) * 5
|
|
usage = int(rng.randint(1, int(3 * difficulty)))
|
|
extra_price = int(rng.randint(1, int(5 * difficulty)))
|
|
|
|
# Ensure price * percent / 100 is an integer
|
|
while (price * percent / 100) != int(price * percent / 100):
|
|
price = int(rng.randint(3, int(25 * difficulty)))
|
|
|
|
result = generate_from_variables(name, item, price, percent, usage,
|
|
extra_item, extra_price, currency, unit)
|
|
|
|
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('Roger', 'coffee', 5, 20, 1, 'donut', 2, '$', 'pound')
|