Merge pull request #63 from open-thought/gsm_symbolic_tests

Gsm symbolic fixes
open-thought · Feb 5, 2025 · aa024ce · aa024ce
2 parents c4c0b3b + 40420a3
commit aa024ce
Show file tree

Hide file tree

Showing 10 changed files with 9,007 additions and 7,360 deletions.
diff --git a/notebooks/verify_gsm_symbolic.ipynb b/notebooks/verify_gsm_symbolic.ipynb
diff --git a/reasoning_gym/arithmetic/__init__.py b/reasoning_gym/arithmetic/__init__.py
@@ -12,8 +12,7 @@
 from .chain_sum import ChainSum, ChainSumConfig
 from .fraction_simplification import FractionSimplificationConfig, FractionSimplificationDataset
 from .gcd import GCDConfig, GCDDataset
-
-# from .gsm_symbolic.gsm_symbolic_datasets import GSMSymbolicDataset, GSMSymbolicDatasetConfig
+from .gsm_symbolic.gsm_symbolic import GSMSymbolicDataset, GSMSymbolicDatasetConfig
 from .lcm import LCMConfig, LCMDataset
 from .leg_counting import LegCountingConfig, LegCountingDataset
 from .prime_factorization import PrimeFactorizationConfig, PrimeFactorizationDataset
@@ -39,8 +38,8 @@
     "LegCountingDataset",
     "PrimeFactorizationConfig",
     "PrimeFactorizationDataset",
-    # "GSMSymbolicDatasetConfig",
-    # "GSMSymbolicDataset",
+    "GSMSymbolicDatasetConfig",
+    "GSMSymbolicDataset",
     "TimeIntervalsConfig",
     "TimeIntervalsDataset",
 ]
diff --git a/reasoning_gym/arithmetic/gsm_symbolic/__init__.py b/reasoning_gym/arithmetic/gsm_symbolic/__init__.py
@@ -0,0 +1,6 @@
+from .gsm_symbolic import GSMSymbolicDataset, GSMSymbolicDatasetConfig
+
+__all__ = [
+    "GSMSymbolicDatasetConfig",
+    "GSMSymbolicDataset",
+]
diff --git a/reasoning_gym/arithmetic/gsm_symbolic/generators.py b/reasoning_gym/arithmetic/gsm_symbolic/generators.py
diff --git a/reasoning_gym/arithmetic/gsm_symbolic/generators_00_49.py b/reasoning_gym/arithmetic/gsm_symbolic/generators_00_49.py