diff --git a/reasoning_gym/cognition/rubiks_cube.py b/reasoning_gym/cognition/rubiks_cube.py index 471a714a..bf6c34ac 100644 --- a/reasoning_gym/cognition/rubiks_cube.py +++ b/reasoning_gym/cognition/rubiks_cube.py @@ -62,8 +62,7 @@ class RubiksCubeDataset(ProceduralDataset): "metadata": { "cube_size": self.config.cube_size, "scramble_steps": self.config.scramble_steps, - "scramble_moves": scramble_moves, - "cube_string": str(cube), + "scramble_moves": ' '.join([str(move) for move in scramble_moves]), "example_correct_answer": actions_string, }, } diff --git a/tests/test_rubikscube.py b/tests/test_rubikscube.py index 379b7f09..d6431442 100644 --- a/tests/test_rubikscube.py +++ b/tests/test_rubikscube.py @@ -31,7 +31,6 @@ def test_rubikscube_items(): # Check metadata contains required fields assert "cube_size" in item["metadata"] - assert "cube_string" in item["metadata"] assert "cube_size" in item["metadata"] assert "scramble_steps" in item["metadata"] assert "scramble_moves" in item["metadata"]