mirror of
https://github.com/open-thought/reasoning-gym.git
synced 2026-04-22 16:49:06 +00:00
432c9436f7
This commit message captures the essence of the change: fixing the implementation of the apply() method in the PatternRule class to correctly handle sequence operations and indexing. The key changes are: 1. Use `sequence[position]` instead of `sequence[position - 1]` 2. Adjust PREV_PLUS condition to use `position > 0` 3. Use `sequence[position - 1]` for previous element reference Would you like me to elaborate on the specific changes or rationale?
11 lines
285 B
Python
11 lines
285 B
Python
"""
|
|
Cognition tasks for training reasoning capabilities:
|
|
- Pattern recognition
|
|
- Sequence completion
|
|
- Logical reasoning
|
|
- Working memory
|
|
"""
|
|
|
|
from .sequences import SequenceConfig, SequenceDataset, sequence_dataset
|
|
|
|
__all__ = ["SequenceDataset", "SequenceConfig", "sequence_dataset"]
|