Create python generator files for gsm symbolic templates

reasoning_gym/arithmetic/gsm_symbolic/generator_81.py

@@ -0,0 +1,56 @@
+from random import Random
+from typing import Dict, Any
+
+def generate_from_variables(occupation: str, weeks_per_month: int, days_per_week: int, 
+                          pay_per_day: int, currency: str) -> Dict[str, Any]:
+    days_per_month = days_per_week * weeks_per_month
+    monthly_pay = days_per_month * pay_per_day
+    yearly_pay = monthly_pay * 12
+
+    question = f"A {occupation} works for {weeks_per_month} weeks every month and for {days_per_week} days every week. If he gets paid {currency}{pay_per_day} every day, how much does he earn if he works for a year?"
+    
+    answer_cot = f"The {occupation} works for {days_per_week} days every week and works for {weeks_per_month} weeks every month so he works for {days_per_week} days/week * {weeks_per_month} weeks/month = {days_per_month} days/month\nIf he earns {currency}{pay_per_day} every day he then earns {currency}{pay_per_day}/day * {days_per_month} days/month = {currency}{monthly_pay}/month\nA year is equal to 12 months so every year he earns {currency}{monthly_pay}/month * 12 months/year = {currency}{yearly_pay}\n#### {yearly_pay}"
+
+    return {
+        'question': question,
+        'answer': str(yearly_pay),
+        'answer_cot': answer_cot,
+        'answer_value': yearly_pay,
+        'variables': {
+            'occupation': occupation,
+            'weeks_per_month': weeks_per_month,
+            'days_per_week': days_per_week,
+            'pay_per_day': pay_per_day,
+            'currency': currency,
+            'days_per_month': days_per_month,
+            'monthly_pay': monthly_pay
+        }
+    }
+
+def generate_example(rng: Random, difficulty: float = 1.0) -> Dict[str, Any]:
+    occupations = ["plumber", "electrician", "painter", "carpenter", "landscaper"]
+    currencies = ["$", "£", "€"]
+    
+    occupation = rng.choice(occupations)
+    currency = rng.choice(currencies)
+    
+    weeks_per_month = int(rng.randint(2, int(5 * difficulty)))
+    days_per_week = int(rng.randint(4, int(7 * difficulty)))
+    pay_per_day = int(rng.randint(40, int(200 * difficulty)) // 5 * 5)
+    
+    result = generate_from_variables(occupation, weeks_per_month, days_per_week, 
+                                   pay_per_day, currency)
+    
+    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("builder", 4, 6, 50, "$")