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

Support float attr

Browse source
This commit is contained in:
EduardDurech 2025-02-09 09:27:57 +00:00
parent 86215b7e5c
commit b8ce5a8a5d
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
reasoning_gym/core/attributes.py
View file

@ -20,7 +20,7 @@ class AttributeDefinition:
default_level: int
description: str
attr_type: AttributeType = AttributeType.STATIC # Default to static
min_value: Optional[int] = None # Minimum value for numeric attributes
min_value: Optional[Union[int, float]] = None # Minimum value for numeric attributes
@classmethod
def validate_attributes(cls, attributes: Dict[str, 'AttributeDefinition'], valid_types: Set[AttributeType], curriculum: str) -> None:
@ -136,6 +136,9 @@ class AttributeDefinition:
# Returns random value up to current level bound, respecting min_value
max_val = self.levels[level]
min_val = self.min_value if self.min_value is not None else 0
# Handle both float and int values
if isinstance(max_val, float) or isinstance(min_val, float):
return lambda: rng.uniform(min_val, max_val)
return lambda: rng.randint(min_val, max_val)
case AttributeType.APPEND: