Create python generator files for gsm symbolic templates

reasoning_gym/arithmetic/gsm_symbolic/generator_98.py

@@ -0,0 +1,78 @@
+from random import Random
+from typing import Dict, Any
+
+def generate_from_variables(name: str, vehicle: str, weight_vehicle: int, 
+                          item: str, weight_item: int, passenger_type: str,
+                          num_passengers: int, weight_passenger: int,
+                          unit: str, force_percent: int) -> Dict[str, Any]:
+    
+    total_passenger_weight = num_passengers * weight_passenger
+    total_weight = weight_vehicle + weight_item + total_passenger_weight
+    force_needed = int((total_weight * force_percent) / 100)
+    
+    question = f"{name}'s {vehicle} breaks down. The {vehicle} weighs {weight_vehicle} {unit} and he has {item} in it weighing {weight_item} {unit}. He also has his {num_passengers} young {passenger_type} who weigh {weight_passenger} {unit} each in it. If the force to move the {vehicle} is {force_percent}% of the weight, how much force does he need to push the {vehicle}?"
+    
+    answer_cot = f"His {num_passengers} {passenger_type} weigh {weight_passenger}*{num_passengers}={total_passenger_weight} {unit}\nSo the total weight of the {vehicle} and everything is {weight_vehicle}+{weight_item}+{total_passenger_weight}={total_weight} {unit}\nSo he needs to generate {total_weight}*{force_percent/100}={force_needed} {unit}\n#### {force_needed}"
+
+    return {
+        'question': question,
+        'answer': str(force_needed),
+        'answer_cot': answer_cot,
+        'answer_value': force_needed,
+        'variables': {
+            'name': name,
+            'vehicle': vehicle,
+            'weight_vehicle': weight_vehicle,
+            'item': item,
+            'weight_item': weight_item,
+            'passenger_type': passenger_type,
+            'num_passengers': num_passengers,
+            'weight_passenger': weight_passenger,
+            'unit': unit,
+            'force_percent': force_percent,
+            'total_weight': total_weight
+        }
+    }
+
+def generate_example(rng: Random, difficulty: float = 1.0) -> Dict[str, Any]:
+    names = ["John", "Michael", "David", "James", "Robert", "William", "Richard"]
+    vehicles = ['car', 'van', 'truck', 'SUV']
+    items = ['luggage', 'groceries', 'equipment', 'furniture']
+    passenger_types = ['children', 'friends', 'colleagues', 'teammates']
+    units = ['pounds', 'kilograms']
+    
+    name = rng.choice(names)
+    vehicle = rng.choice(vehicles)
+    item = rng.choice(items)
+    passenger_type = rng.choice(passenger_types)
+    unit = rng.choice(units)
+    
+    weight_vehicle = int(rng.randint(1000, int(3000 * difficulty) // 50) * 50)
+    weight_item = int(rng.randint(100, int(500 * difficulty) // 25) * 25)
+    weight_passenger = int(rng.randint(50, int(100 * difficulty) // 5) * 5)
+    num_passengers = int(rng.randint(2, int(5 * difficulty)))
+    force_percent = int(rng.randint(1, int(6 * difficulty)))
+    
+    # Ensure force calculation results in integer
+    total_weight = weight_vehicle + weight_item + (num_passengers * weight_passenger)
+    while (total_weight * force_percent) % 100 != 0:
+        force_percent = int(rng.randint(1, int(6 * difficulty)))
+    
+    result = generate_from_variables(name, vehicle, weight_vehicle, item, weight_item,
+                                   passenger_type, num_passengers, weight_passenger,
+                                   unit, force_percent)
+    
+    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", "car", 1200, "luggage", 250,
+                                 "children", 2, 75, "pounds", 1)