Create python generator files for gsm symbolic data templates

.gitignore

@@ -40,3 +40,7 @@ htmlcov/
 
 # OSX
 .DS_Store
+
+# Jupyter Notebook
+.ipynb_checkpoints/
+.virtual_documents/

notebooks/test_generator_files.py

@@ -0,0 +1,53 @@
+import os
+import importlib.util
+from datetime import datetime
+from typing import Dict, Any
+
+def test_generator_files(directory_path: str) -> None:
+    """Test all generator files in directory"""
+
+    # Setup logging
+    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
+    log_file = f"generator_tests_{timestamp}.log"
+
+    with open(log_file, 'w') as log:
+        # Walk through directory
+        for root, _, files in os.walk(directory_path):
+            for file in files:
+                if not file.endswith('.py'):
+                    continue
+
+                filepath = os.path.join(root, file)
+                try:
+                    # Import module dynamically
+                    spec = importlib.util.spec_from_file_location(
+                        file[:-3], filepath
+                    )
+                    module = importlib.util.module_from_spec(spec)
+                    spec.loader.exec_module(module)
+
+                    # Check for original_example function
+                    if hasattr(module, 'original_example'):
+                        try:
+                            result = module.original_example()
+                            if isinstance(result, dict):
+                                message = f"SUCCESS: {file} - returned valid dictionary\n"
+                            else:
+                                message = f"ERROR: {file} - did not return dictionary\n"
+                        except Exception as e:
+                            message = f"ERROR: {file} - execution failed: {str(e)}\n"
+                    else:
+                        message = f"ERROR: {file} - no original_example() found\n"
+
+                except Exception as e:
+                    message = f"ERROR: {file} - import failed: {str(e)}\n"
+
+                # Log result
+                log.write(message)
+                print(message, end='')
+
+# Usage
+if __name__ == "__main__":
+    generator_path = "../reasoning_gym/arithmetic/gsm_symbolic"
+    test_generator_files(generator_path)
+

reasoning_gym/arithmetic/__init__.py

@@ -15,6 +15,7 @@
 from .lcm import LCMConfig, LCMDataset
 from .leg_counting import LegCountingConfig, LegCountingDataset
 from .prime_factorization import PrimeFactorizationConfig, PrimeFactorizationDataset
+from .gsm_symbolic.gsm_symbolic_datasets import GSMSymbolicDataset, GSMSymbolicDatasetConfig
 from .time_intervals import TimeIntervalsConfig, TimeIntervalsDataset
 
 __all__ = [
@@ -37,6 +38,8 @@
     "LegCountingDataset",
     "PrimeFactorizationConfig",
     "PrimeFactorizationDataset",
+    "GSMSymbolicDatasetConfig",
+    "GSMSymbolicDataset",
     "TimeIntervalsConfig",
     "TimeIntervalsDataset",
 ]