Create python generator files for gsm symbolic templates

reasoning_gym/arithmetic/gsm_symbolic/generator_32.py

@@ -0,0 +1,77 @@
+from random import Random
+from typing import Dict, Any
+
+def generate_from_variables(name: str, periods: int, extra_classes: int, 
+                          mins_per_class: int, days: int, weekend_fraction: float) -> Dict[str, Any]:
+    total_classes = periods + extra_classes
+    daily_mins = total_classes * mins_per_class  
+    weekly_mins = daily_mins * days
+    weekend_mins = int(weekly_mins * weekend_fraction)
+    total_mins = weekly_mins + 2 * weekend_mins
+    total_hours = total_mins // 60
+
+    question = f"There are {periods} periods in the day for a normal student but {name} has to take {extra_classes} extra classes. Each class is {mins_per_class} minutes long. He goes to class for {days} days a week. He then spends {weekend_fraction} of his weekly minutes each on Saturday and Sunday as extra learning time. How many hours a week does he spend learning?"
+
+    answer_cot = f"He takes {periods}+{extra_classes}={total_classes} classes a day\n" \
+                 f"That means he spends {mins_per_class}*{total_classes}={daily_mins} minutes per day in class\n" \
+                 f"So he spends {daily_mins}*{days}={weekly_mins} minutes a week\n" \
+                 f"That means he spends {weekly_mins}*{weekend_fraction}={weekend_mins} minutes each on Saturday and Sunday\n" \
+                 f"So he spends {weekly_mins}+{weekend_mins}+{weekend_mins}={total_mins} minutes per week\n" \
+                 f"So he spends {total_mins}/60={total_hours} hours per week\n#### {total_hours}"
+
+    return {
+        'question': question,
+        'answer': str(total_hours),
+        'answer_cot': answer_cot,
+        'answer_value': total_hours,
+        'variables': {
+            'name': name,
+            'periods': periods,
+            'extra_classes': extra_classes,
+            'mins_per_class': mins_per_class,
+            'days': days,
+            'weekend_fraction': weekend_fraction,
+            'total_classes': total_classes,
+            'daily_mins': daily_mins,
+            'weekly_mins': weekly_mins,
+            'weekend_mins': weekend_mins,
+            'total_mins': total_mins
+        }
+    }
+
+def generate_example(rng: Random, difficulty: float = 1.0) -> Dict[str, Any]:
+    names = ["John", "James", "William", "Michael", "David", "Richard", "Thomas"]
+    fractions = ["1/16", "1/8", "1/4", "1/2"]
+    
+    name = rng.choice(names)
+    periods = int(rng.randint(5, int(10 * difficulty)))
+    extra_classes = int(rng.randint(1, int(5 * difficulty)))
+    mins_per_class = int(rng.randint(30, int(61 * difficulty)) // 5 * 5)
+    days = int(rng.randint(4, int(7 * difficulty)))
+    weekend_fraction = float(eval(rng.choice(fractions)))
+    
+    # Ensure results are integers
+    while not (((periods + extra_classes) * mins_per_class * days * weekend_fraction).is_integer() and
+              (((periods + extra_classes) * mins_per_class * days + 
+                2 * (periods + extra_classes) * mins_per_class * days * weekend_fraction) / 60).is_integer()):
+        periods = int(rng.randint(5, int(10 * difficulty)))
+        extra_classes = int(rng.randint(1, int(5 * difficulty)))
+        mins_per_class = int(rng.randint(30, int(61 * difficulty)) // 5 * 5)
+        days = int(rng.randint(4, int(7 * difficulty)))
+    
+    result = generate_from_variables(name, periods, extra_classes, mins_per_class, 
+                                   days, weekend_fraction)
+    
+    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("John", 6, 2, 40, 5, 1/16)