From a9ea2df7a0594d152aa2b26b175a824c3a25bdf4 Mon Sep 17 00:00:00 2001 From: joesharratt1229 Date: Sun, 2 Feb 2025 15:28:52 +0000 Subject: [PATCH] added register dataset to script --- reasoning_gym/algebra/intermediate_integration.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/reasoning_gym/algebra/intermediate_integration.py b/reasoning_gym/algebra/intermediate_integration.py index df0388a8..9e5c9528 100644 --- a/reasoning_gym/algebra/intermediate_integration.py +++ b/reasoning_gym/algebra/intermediate_integration.py @@ -1,7 +1,6 @@ import random from dataclasses import dataclass -from fractions import Fraction -from typing import List, Optional +from typing import Optional import sympy @@ -232,3 +231,6 @@ class IntermediateIntegrationDataset(ProceduralDataset): "type": substitution_type if problem_type == "substitution" else parts_type, }, } + + +register_dataset("intermediate_integration", IntermediateIntegrationDataset, IntermediateIntegrationConfig)