open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-19 12:58:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

added validation

Browse source
This commit is contained in:
joesharratt1229 2025-02-08 10:21:04 +00:00
parent 4c94975d15
commit 4d9ea46c85
1 changed files with 4 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

4
reasoning_gym/arc/rearc.py
View file

@ -58,6 +58,10 @@ class ReArcConfig:
seed: Optional[int] = None
size: int = 500
def validate(self):
assert self.diff_lb < self.diff_ub, "diff_lb must be < diff_ub."
assert self.size > 0, "Size of dataset must be positive."
class ReArcDataset(ProceduralDataset):
def __init__(self, config: ReArcConfig):