reasoning-gym/reasoning_gym/cognition/__init__.py
2025-01-24 16:23:02 +01:00

11 lines
316 B
Python

"""
Cognition tasks for training reasoning capabilities:
- Pattern recognition
- Sequence completion
- Logical reasoning
- Working memory
"""
from .number_sequences import NumberSequenceConfig, NumberSequenceDataset, sequence_dataset
__all__ = ["NumberSequenceDataset", "NumberSequenceConfig", "sequence_dataset"]