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

fix rubiks cube prompt (#433)

Browse source
This commit is contained in:
Zafir Stojanovski 2025-05-15 22:31:21 +02:00 • committed by GitHub
parent 85f3c6dd02
commit 6d9e14729a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
reasoning_gym/cognition/rubiks_cube.py
View file

@ -39,7 +39,7 @@ class RubiksCubeDataset(ProceduralDataset):
def __init__(self, config: RubiksCubeConfig):
self._prompt_templates = [
"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. Do not combine any steps, for instance, do not write 'U2', and instead write 'U U'.",
"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 using Singmaster notation. Do not combine any steps, for instance, do not write 'U2', and instead write 'U U'.",
"You see a size {cube_size} Rubik's cube arranged as follows::\n\n{cube_render} \n\nPlease provide a solution to solve this cube using Singmaster notation. Do not combine any steps, for instance, do not write 'U2', and instead write 'U U'.",
]
super().__init__(config=config, seed=config.seed, size=config.size)