diff --git a/reasoning_gym/cognition/rubiks_cube.py b/reasoning_gym/cognition/rubiks_cube.py index b82e4882..e2cbf4e1 100644 --- a/reasoning_gym/cognition/rubiks_cube.py +++ b/reasoning_gym/cognition/rubiks_cube.py @@ -1,16 +1,12 @@ from dataclasses import dataclass import random import re - from magiccube.cube import Cube from magiccube.solver.basic.basic_solver import BasicSolver - from typing import List, Optional, Tuple, Dict - from ..factory import ProceduralDataset, register_dataset - @dataclass class RubiksCubeConfig: """Configuration for RubiksCube task generation""" diff --git a/tests/test_rubikscube.py b/tests/test_rubikscube.py index d6431442..a49e877d 100644 --- a/tests/test_rubikscube.py +++ b/tests/test_rubikscube.py @@ -39,7 +39,3 @@ def test_rubikscube_items(): assert dataset.score_answer(answer=item['metadata']['example_correct_answer'], entry=item) == 1.0 assert dataset.score_answer(answer='R', entry=item) == 0.01 assert dataset.score_answer(answer=None, entry=item) == 0.0 - - - -