Create python generator files for gsm symbolic templates

2026-04-25 17:10:51 +00:00 · 2025-02-01 20:23:30 -06:00 · 2025-02-01 20:23:30 -06:00 · 1e0dbc9875
commit 1e0dbc9875
parent ff13dc6825
100 changed files with 6919 additions and 0 deletions
--- a/reasoning_gym/arithmetic/gsm_symbolic/generator_35.py
+++ b/reasoning_gym/arithmetic/gsm_symbolic/generator_35.py
@ -0,0 +1,79 @@
+from random import Random
+from typing import Dict, Any
+
+def generate_from_variables(job: str, building: str, room: str, num_rooms: int, 
+                          num_days: int, time_per_room: int, hours_per_day: int) -> Dict[str, Any]:
+    
+    rooms_per_day = num_rooms // num_days
+    minutes_per_day = rooms_per_day * time_per_room
+    hours_cleaning = minutes_per_day / 60
+    percentage = int(100 * hours_cleaning / hours_per_day)
+
+    question = f"A {job} has to clean a {building} with {num_rooms} {room}s. They have {num_days} days to get it done. It takes them {time_per_room} minutes per {room}. If they work {hours_per_day} hour day, what percentage of their day, on average, is spent cleaning {room}s?"
+    
+    answer_cot = f"They have to clean {rooms_per_day} {room}s a day because {num_rooms} / {num_days} = {rooms_per_day}\n" \
+                 f"They spend {minutes_per_day} minutes cleaning per day because {rooms_per_day} x {time_per_room} = {minutes_per_day}\n" \
+                 f"They spend {hours_cleaning} hours a day because {minutes_per_day} / 60 = {hours_cleaning}\n" \
+                 f"They spend {hours_cleaning/hours_per_day} of their day cleaning {room}s because {hours_cleaning} / {hours_per_day} = {hours_cleaning/hours_per_day}\n" \
+                 f"They spend {percentage}% of their day cleaning {room}s because {hours_cleaning/hours_per_day} x 100 = {percentage}\n" \
+                 f"#### {percentage}"
+
+    return {
+        'question': question,
+        'answer': str(percentage),
+        'answer_cot': answer_cot,
+        'answer_value': percentage,
+        'variables': {
+            'job': job,
+            'building': building,
+            'room': room,
+            'num_rooms': num_rooms,
+            'num_days': num_days,
+            'time_per_room': time_per_room,
+            'hours_per_day': hours_per_day,
+            'rooms_per_day': rooms_per_day,
+            'minutes_per_day': minutes_per_day,
+            'hours_cleaning': hours_cleaning,
+            'percentage': percentage
+        }
+    }
+
+def generate_example(rng: Random, difficulty: float = 1.0) -> Dict[str, Any]:
+    jobs = ["janitor", "cleaner", "maintenance worker"]
+    buildings = ["office building", "hospital", "university"]
+    rooms = ["room", "floor"]
+    
+    job = rng.choice(jobs)
+    building = rng.choice(buildings)
+    room = rng.choice(rooms)
+
+    num_days = int(rng.randint(3, int(12 * difficulty)))
+    time_per_room = int(rng.randrange(10, int(46 * difficulty), 5))
+    hours_per_day = int(rng.randint(6, int(17 * difficulty)))
+    
+    # Generate num_rooms ensuring divisibility by num_days
+    rooms_per_day = rng.randint(5, int(20 * difficulty))
+    num_rooms = rooms_per_day * num_days
+    
+    # Ensure conditions are met
+    while (num_rooms/num_days) * time_per_room >= hours_per_day * 60:
+        rooms_per_day = rng.randint(5, int(20 * difficulty))
+        num_rooms = rooms_per_day * num_days
+
+    result = generate_from_variables(job, building, room, num_rooms, num_days,
+                                   time_per_room, hours_per_day)
+    
+    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("custodian", "school", "classroom", 
+                                 80, 5, 15, 8)