open-thought/reasoning-gym
1
0
Fork 0
mirror of https://github.com/open-thought/reasoning-gym.git synced 2026-04-28 17:29:39 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Template Abstraction

Browse source
This commit is contained in:
EduardDurech 2025-02-07 17:06:50 +00:00
parent 227319f1da
commit c2e3dbc826
8 changed files with 325 additions and 387 deletions
Download patch file Download diff file Expand all files Collapse all files

6
reasoning_gym/core/attributes.py
View file

@ -12,6 +12,7 @@ class AttributeType(Enum):
STATIC = "static" # Each level is independent
UBOUND = "ubound" # Each level is an upper bound
APPEND = "append" # Each level includes all previous levels
APPEND_LIST = "append_list" # Each level includes all previous levels
@dataclass
class AttributeDefinition:
@ -143,4 +144,9 @@ class AttributeDefinition:
available_values = self.levels[:level + 1]
return lambda: rng.choice(available_values)
case AttributeType.APPEND_LIST:
# Returns random choice from accumulated values up to current level
available_values = sum(self.levels[:level + 1], [])
return lambda: rng.choice(available_values)
raise ValueError(f"Unknown attribute type: {self.attr_type} for attribute '{self.description}'")