mirror of
https://github.com/open-thought/reasoning-gym.git
synced 2026-04-25 17:10:51 +00:00
lint
This commit is contained in:
parent
a872a09030
commit
9c4e820fc4
2 changed files with 0 additions and 8 deletions
|
|
@ -1,16 +1,12 @@
|
||||||
from dataclasses import dataclass
|
from dataclasses import dataclass
|
||||||
import random
|
import random
|
||||||
import re
|
import re
|
||||||
|
|
||||||
from magiccube.cube import Cube
|
from magiccube.cube import Cube
|
||||||
from magiccube.solver.basic.basic_solver import BasicSolver
|
from magiccube.solver.basic.basic_solver import BasicSolver
|
||||||
|
|
||||||
from typing import List, Optional, Tuple, Dict
|
from typing import List, Optional, Tuple, Dict
|
||||||
|
|
||||||
|
|
||||||
from ..factory import ProceduralDataset, register_dataset
|
from ..factory import ProceduralDataset, register_dataset
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
@dataclass
|
||||||
class RubiksCubeConfig:
|
class RubiksCubeConfig:
|
||||||
"""Configuration for RubiksCube task generation"""
|
"""Configuration for RubiksCube task generation"""
|
||||||
|
|
|
||||||
|
|
@ -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=item['metadata']['example_correct_answer'], entry=item) == 1.0
|
||||||
assert dataset.score_answer(answer='R', entry=item) == 0.01
|
assert dataset.score_answer(answer='R', entry=item) == 0.01
|
||||||
assert dataset.score_answer(answer=None, entry=item) == 0.0
|
assert dataset.score_answer(answer=None, entry=item) == 0.0
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue