From cdcd79ae2be63100fe06f759eb1f63a7bd5a2248 Mon Sep 17 00:00:00 2001 From: vncntt Date: Wed, 19 Feb 2025 15:30:30 -0800 Subject: [PATCH] register decimal_chain_sum --- reasoning_gym/arithmetic/decimal_chain_sum.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/reasoning_gym/arithmetic/decimal_chain_sum.py b/reasoning_gym/arithmetic/decimal_chain_sum.py index eaf5c7ea..da920c9d 100644 --- a/reasoning_gym/arithmetic/decimal_chain_sum.py +++ b/reasoning_gym/arithmetic/decimal_chain_sum.py @@ -152,3 +152,6 @@ class DecimalChainSumDataset(ProceduralDataset): return 1.0 if student_answer == oracle_answer else 0.01 except (ValueError, TypeError, ArithmeticError): return 0.01 + + +register_dataset("decimal_chain_sum", DecimalChainSumDataset, DecimalChainSumConfig)