Create python generator files for gsm symbolic templates

reasoning_gym/arithmetic/gsm_symbolic/generator_38.py

@@ -0,0 +1,67 @@
+from random import Random
+from typing import Dict, Any
+
+def generate_from_variables(name: str, count: int, child_type: str,
+                          item1: str, item2: str, item3: str, item4: str, item5: str,
+                          n1: int, n2: int, n3: int, n4: int, n5: int) -> Dict[str, Any]:
+    
+    skeins_per_child = n1 + n2 + n3 + n4 + n5
+    total_skeins = count * skeins_per_child
+
+    question = f"{name} is knitting winter wear for her {count} grandchildren. They're {child_type}, so they're all the same size. She wants to make a {item1}, {item2}, {item3}, {item4}, and {item5} for each of them. It takes {n1} skeins of wool to make a {item1}, {n2} for a {item2}, {n3} for a {item3}, {n4} for a pair of {item4}, and {n5} for a pair of {item5}. How many skeins of wool will she need to buy?"
+
+    answer_cot = f"A full outfit for each child will require {n1} skeins per {item1} + {n2} skeins per {item2} + {n3} skeins per {item3} + {n4} skeins per pair of {item4} + {n5} skeins per pair of {item5} = {skeins_per_child} skeins of wool.\nSo to knit outfits for all of her grandchildren, she will need {count} * {skeins_per_child} = {total_skeins} skeins of wool.\n#### {total_skeins}"
+
+    return {
+        'question': question,
+        'answer': str(total_skeins),
+        'answer_cot': answer_cot,
+        'answer_value': total_skeins,
+        'variables': {
+            'name': name,
+            'count': count,
+            'child_type': child_type,
+            'items': [item1, item2, item3, item4, item5],
+            'skeins_per_item': [n1, n2, n3, n4, n5],
+            'skeins_per_child': skeins_per_child,
+            'total_skeins': total_skeins
+        }
+    }
+
+def generate_example(rng: Random, difficulty: float = 1.0) -> Dict[str, Any]:
+    names_female = ["Martha", "Mary", "Elizabeth", "Susan", "Margaret", "Patricia"]
+    clothing_items = ["sweater", "shawl", "hat", "cardigan", "poncho", "vest", "beanie", "tunic"]
+    accessories = ["mittens", "booties", "socks", "leg warmers", "gloves"]
+    children_types = [("twins", 2), ("triplets", 3), ("quadruplets", 4), ("quintuplets", 5)]
+
+    name = rng.choice(names_female)
+    child_type, count = rng.choice(children_types)
+    item1, item2, item3 = rng.sample(clothing_items, 3)
+    item4, item5 = rng.sample(accessories, 2)
+
+    # Scale numbers based on difficulty
+    n1 = int(rng.randint(3, int(19 * difficulty)))
+    n2 = int(rng.randint(3, int(19 * difficulty)))
+    n3 = int(rng.randint(3, int(19 * difficulty)))
+    n4 = int(rng.randint(3, int(19 * difficulty)))
+    n5 = int(rng.randint(3, int(19 * difficulty)))
+
+    result = generate_from_variables(name, count, child_type,
+                                   item1, item2, item3, item4, item5,
+                                   n1, n2, n3, n4, n5)
+
+    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("Martha", 3, "triplets",
+                                 "hat", "scarf", "sweater", "mittens", "socks",
+                                 2, 4, 12, 1, 2)