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

cleanup

Browse source
This commit is contained in:
Rich Jones 2025-01-29 16:07:27 +01:00
parent ebcd6d5e43
commit cc69d90a4f
1 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
reasoning_gym/cognition/rubiks_cube.py
View file

@ -46,7 +46,6 @@ class RubiksCubeDataset(ProceduralDataset):
cube = Cube(self.config.cube_size)
scramble_moves = cube.scramble(num_steps=self.config.scramble_steps)
cube_string = cube.cube
cube_render = self.remove_ansi(str(cube))
if self.config.cube_size == 3:
@ -57,7 +56,7 @@ class RubiksCubeDataset(ProceduralDataset):
actions = None
return {
"question": random.choice(self._prompt_templates).format(cube_size=self.config.cube_size, cube_string=cube_string, cube_render=cube_render),
"question": random.choice(self._prompt_templates).format(cube_size=self.config.cube_size, cube_render=cube_render),
"answer": None,
"metadata": {
"cube_size": self.config.cube_size,