Create python generator files for gsm symbolic templates

reasoning_gym/arithmetic/gsm_symbolic/generator_21.py

@@ -0,0 +1,72 @@
+from random import Random
+from typing import Dict, Any
+
+def generate_from_variables(name: str, length: int, unit_length: str, plant_width: int,
+                          space: float, owned: int, currency: str, cost: int) -> Dict[str, Any]:
+    total_plants = int(length / space)
+    plants_to_buy = total_plants - owned
+    total_cost = plants_to_buy * cost
+
+    question = f"{name} has a flower bed that is {length} {unit_length} long. {name} wants to fill her flower bed with plants. {name}'s flowers grow {plant_width} inches wide so she needs to leave {space} {unit_length} between every plant. {name} already owns {owned} flowers. Each flowering plant costs {currency}{cost} at the store, how much money will {name} spend at the store to fill up her flower bed?"
+    
+    answer_cot = f"{name}'s flower bed is {length} {unit_length} / {space} {unit_length} per plant = {total_plants} plants needed.\n{name} needs to buy {total_plants} plants - {owned} plants = {plants_to_buy} plants needed to purchase.\n{name} will spend {plants_to_buy} plants * {currency}{cost} = {currency}{total_cost}.\n#### {total_cost}"
+
+    return {
+        'question': question,
+        'answer': str(total_cost),
+        'answer_cot': answer_cot,
+        'answer_value': total_cost,
+        'variables': {
+            'name': name,
+            'bed_length': length,
+            'unit': unit_length,
+            'plant_width': plant_width,
+            'plant_spacing': space,
+            'owned_plants': owned,
+            'currency': currency,
+            'cost_per_plant': cost,
+            'total_plants': total_plants,
+            'plants_to_buy': plants_to_buy
+        }
+    }
+
+def generate_example(rng: Random, difficulty: float = 1.0) -> Dict[str, Any]:
+    names_female = ['Emma', 'Olivia', 'Ava', 'Isabella', 'Sophia', 'Mia', 'Charlotte']
+    currencies = ['$', '£', '€']
+    units = ['feet', 'meters']
+    
+    name = rng.choice(names_female)
+    unit = rng.choice(units)
+    currency = rng.choice(currencies)
+    
+    length = int(rng.randint(110, int(220 * difficulty)))
+    plant_width = int(rng.randint(2, int(8 * difficulty)))
+    space = round(rng.uniform(1.25, 2.0) * difficulty, 2)
+    owned = int(rng.randint(10, int(30 * difficulty)))
+    cost = int(rng.randint(3, int(15 * difficulty)))
+
+    # Ensure conditions are met
+    while not (plant_width * 3 < length and 
+              plant_width < space and
+              length % space == 0 and
+              length / space > owned):
+        length = int(rng.randint(110, int(220 * difficulty)))
+        space = round(rng.uniform(1.25, 2.0) * difficulty, 2)
+        owned = int(rng.randint(10, int(30 * difficulty)))
+
+    result = generate_from_variables(name, length, unit, plant_width, space, owned, 
+                                   currency, cost)
+    
+    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('Pat', 111, 'feet', 12, 1.5, 17, '$', 6)