open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-30 17:40:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

added register dataset to script

Browse source
This commit is contained in:
joesharratt1229 2025-02-02 15:28:52 +00:00
parent ed1492ba05
commit a9ea2df7a0
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
reasoning_gym/algebra/intermediate_integration.py
View file

@ -1,7 +1,6 @@
import random import random
from dataclasses import dataclass from dataclasses import dataclass
from fractions import Fraction from typing import Optional
from typing import List, Optional
import sympy import sympy
@ -232,3 +231,6 @@ class IntermediateIntegrationDataset(ProceduralDataset):
"type": substitution_type if problem_type == "substitution" else parts_type, "type": substitution_type if problem_type == "substitution" else parts_type,
}, },
} }
register_dataset("intermediate_integration", IntermediateIntegrationDataset, IntermediateIntegrationConfig)