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

refactor: Reorganize ArcAgiConfig class attributes for better readability

Browse source
This commit is contained in:
Andreas Koepf 2025-02-09 00:12:59 +01:00 • committed by Andreas Koepf (aider)
parent e56316ebb2
commit b73040b066
1 changed files with 5 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

7
reasoning_gym/arc/arc_agi.py
View file

@ -20,11 +20,14 @@ class ArcAgiConfig:
use_train: bool = True
use_eval: bool = True
board_format_opts: BoardFormattingOptions = field(default_factory=lambda: BoardFormattingOptions())
seed: Optional[int] = None
size: int = 500
# Augmentation options
use_rotations: bool = True
use_mirrors: bool = True
use_color_permutation: bool = True
seed: Optional[int] = None
size: int = 500
def validate(self):
assert self.size > 0, "Size of dataset must be positive."