Create python generator files for gsm symbolic templates

reasoning_gym/arithmetic/gsm_symbolic/generator_25.py

@@ -0,0 +1,77 @@
+from random import Random
+from typing import Dict, Any
+from fractions import Fraction
+
+def generate_from_variables(food: str, calories: int, size: int, servings: int, 
+                          total_target: int, consumed: int, unit: str) -> Dict[str, Any]:
+    
+    calories_left = total_target - consumed
+    serving_fraction = Fraction(calories_left, calories)
+    grams_per_serving = size // servings
+    grams_allowed = grams_per_serving * serving_fraction
+
+    question = f"According to its nutritional info, a bag of {food} has {calories} calories per serving. If a {size} {unit} bag has {servings} servings, how many {unit} can you eat if your daily calorie target is {total_target} and you have already consumed {consumed} calories?"
+
+    answer_cot = f"If the total calorie target is {total_target} and I have consumed {consumed} calories then I have {total_target}-{consumed} = {calories_left} calories left to eat\n" \
+                 f"If each serving of {food} has {calories} calories and I only have {calories_left} calories left to eat, then I can only eat {calories_left}/{calories} of a serving = {serving_fraction} of a serving\n" \
+                 f"We also know that a {size} {unit} bag of {food} has {servings} servings, hence each serving has {size} {unit}/{servings} = {grams_per_serving} {unit}\n" \
+                 f"If I can only eat {serving_fraction} of a serving, then I can eat only {grams_per_serving} * {serving_fraction} = {grams_allowed} {unit}\n" \
+                 f"#### {float(grams_allowed)}"
+
+    return {
+        'question': question,
+        'answer': str(float(grams_allowed)),
+        'answer_cot': answer_cot,
+        'answer_value': float(grams_allowed),
+        'variables': {
+            'food': food,
+            'calories': calories,
+            'size': size,
+            'servings': servings,
+            'total_target': total_target,
+            'consumed': consumed,
+            'unit': unit,
+            'calories_left': calories_left,
+            'grams_per_serving': grams_per_serving,
+            'serving_fraction': serving_fraction
+        }
+    }
+
+def generate_example(rng: Random, difficulty: float = 1.0) -> Dict[str, Any]:
+    foods = ["popcorn", "breads", "cookies"]
+    units = ["grams", "ounces", "oz"]
+    
+    food = rng.choice(foods)
+    unit = rng.choice(units)
+    
+    calories = int(rng.randint(150, int(500 * difficulty)) // 25 * 25)
+    size = int(rng.randint(100, int(400 * difficulty)) // 25 * 25)
+    servings = int(rng.randint(4, int(8 * difficulty)))
+    total_target = int(rng.randint(1900, int(2500 * difficulty)) // 5 * 5)
+    consumed = int(rng.randint(600, int(1800 * difficulty)) // 25 * 25)
+
+    # Ensure conditions are met
+    while (consumed >= total_target or 
+           not (size % servings == 0) or
+           not ((size//servings) * Fraction(total_target-consumed, calories)).denominator == 1):
+        calories = int(rng.randint(150, int(500 * difficulty)) // 25 * 25)
+        size = int(rng.randint(100, int(400 * difficulty)) // 25 * 25)
+        servings = int(rng.randint(4, int(8 * difficulty)))
+        total_target = int(rng.randint(1900, int(2500 * difficulty)) // 5 * 5)
+        consumed = int(rng.randint(600, int(1800 * difficulty)) // 25 * 25)
+
+    result = generate_from_variables(food, calories, size, servings, total_target, consumed, 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("chips", 250, 300, 5, 2000, 1800, "grams")