open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-23 16:55:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix index out of range of arc_1d dataset (#190)

Browse source
This commit is contained in:
Andreas Koepf 2025-02-23 12:51:41 +01:00
parent e4102a44f6
commit 7a45b14a49
3 changed files with 30 additions and 24 deletions
Download patch file Download diff file Expand all files Collapse all files

2
reasoning_gym/arc/arc_1d.py
View file

@ -18,7 +18,7 @@ class Arc1DConfig:
def validate(self) -> None:
"""Validate configuration parameters"""
assert self.min_size > 0, "min_size must be positive"
assert self.min_size >= 8, "min_size must be >= 8"
assert self.max_size >= self.min_size, "max_size must be >= min_size"
assert self.num_train > 0, "num_train must be positive"
assert self.size > 0, "size must be positive"