mirror of
https://github.com/open-thought/reasoning-gym.git
synced 2026-04-29 17:35:16 +00:00
fix unit tests, lower python dependency to 3.9
This commit is contained in:
parent
ee67374aae
commit
ad9f0d265c
11 changed files with 66 additions and 56 deletions
|
|
@ -1,18 +1,18 @@
|
|||
import random
|
||||
from dataclasses import dataclass
|
||||
from enum import StrEnum
|
||||
from enum import Enum
|
||||
from itertools import count
|
||||
from typing import Dict, List, Optional, Set, Tuple
|
||||
|
||||
from ..factory import ProceduralDataset, register_dataset
|
||||
|
||||
|
||||
class Gender(StrEnum):
|
||||
class Gender(Enum):
|
||||
MALE = "male"
|
||||
FEMALE = "female"
|
||||
|
||||
|
||||
class Relationship(StrEnum):
|
||||
class Relationship(Enum):
|
||||
MOTHER = "mother"
|
||||
FATHER = "father"
|
||||
SISTER = "sister"
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue