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

add seed & size params to RubicsCubeConfig

Browse source
This commit is contained in:
Andreas Koepf 2025-01-30 00:49:41 +01:00
parent 08ee435d47
commit 25505e3a75
2 changed files with 11 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

4
reasoning_gym/cognition/rubiks_cube.py
View file

@ -14,6 +14,8 @@ class RubiksCubeConfig:
scramble_steps: int = 3 # Number of random steps from initial state
cube_size: int = 3 # Default to a standard 3x3x3 cube
seed: Optional[int] = None
size: int = 500
def validate(self) -> None:
"""Validate configuration parameters"""
@ -30,7 +32,7 @@ class RubiksCubeDataset(ProceduralDataset):
"You are given a {cube_size}x{cube_size}x{cube_size} Rubik's cube. It looks like this:\n\n{cube_render} \n\nPlease provide a solution to solve this cube using Singmaster notation.",
"You see a size {cube_size} Rubik's cube. It is arranged this:\n\n{cube_render} \n\nPlease provide a solution to solve this cube.",
]
super().__init__(config=config)
super().__init__(config=config, seed=config.seed, size=config.size)
def _generate_random_moves(self, rng: Random, cube: Cube, num_steps: int = 50, wide=None) -> List[CubeMove]:
"""Generate a list of random moves (but don't apply them).