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

formatting

Browse source
This commit is contained in:
Andreas Koepf 2025-01-24 10:34:07 +01:00
parent 98988c8481
commit 20069b2a7d
37 changed files with 504 additions and 666 deletions
Download patch file Download diff file Expand all files Collapse all files

6
reasoning_gym/cognition/sequences.py
View file

@ -40,7 +40,7 @@ class SequenceConfig:
class PatternRule:
"""Represents a composable sequence pattern rule"""
def __init__(self, operations: List[Operation], parameters: List[int], subrules: List['PatternRule'] = None):
def __init__(self, operations: List[Operation], parameters: List[int], subrules: List["PatternRule"] = None):
self.operations = operations
self.parameters = parameters
self.subrules = subrules or []
@ -66,14 +66,14 @@ class PatternRule:
elif op == Operation.COMPOSE:
# Apply each subrule in sequence, passing the result through
for subrule in self.subrules:
temp_sequence = sequence[:position + 1]
temp_sequence = sequence[: position + 1]
temp_sequence[-1] = result # Use current result as input
result = subrule.apply(temp_sequence, position)
return result
@classmethod
def compose(cls, rules: List['PatternRule']) -> 'PatternRule':
def compose(cls, rules: List["PatternRule"]) -> "PatternRule":
"""Create a new rule that composes multiple rules together"""
return cls([Operation.COMPOSE], [0], subrules=rules)