From 285e2b20cce00a3c3691571fa425347d4b0b420f Mon Sep 17 00:00:00 2001 From: Rich Jones Date: Wed, 26 Feb 2025 13:07:17 +0100 Subject: [PATCH] rubiks cube instructions --- reasoning_gym/cognition/rubiks_cube.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/reasoning_gym/cognition/rubiks_cube.py b/reasoning_gym/cognition/rubiks_cube.py index c34fd30a..f39d6176 100644 --- a/reasoning_gym/cognition/rubiks_cube.py +++ b/reasoning_gym/cognition/rubiks_cube.py @@ -31,8 +31,8 @@ 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.", - "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.", + "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'.", ] super().__init__(config=config, seed=config.seed, size=config.size)