Commit graph

12 commits

Author SHA1 Message Date
Andreas Koepf (aider)
95fdca5a92 feat: Add special case handling for min_digits=1 in ChainSum generation 2025-01-23 12:07:56 +01:00
Andreas Koepf (aider)
e1734b0b11 refactor: Move min_value and max_value calculations to __getitem__ 2025-01-23 12:05:55 +01:00
Andreas Koepf (aider)
1ff01627cc feat: Implement allow_negation to generate both positive and negative numbers in ChainSum 2025-01-23 12:01:21 +01:00
Andreas Koepf
358829cc60 refactor: Replace Random import with random module and update type hints 2025-01-23 12:01:20 +01:00
Andreas Koepf (aider)
654bed3146 refactor: Simplify ChainSum random number generation with base seed 2025-01-23 11:56:36 +01:00
Andreas Koepf (aider)
d46f338160 feat: Add digit-based number range generation for chain sum tasks 2025-01-23 11:46:55 +01:00
Andreas Koepf (aider)
3d9b48ba31 feat: Add ChainSum class for generating simple arithmetic tasks 2025-01-23 11:40:00 +01:00
Andreas Koepf
37338d2107 chore: Add __init__.py to arithmetic module 2025-01-23 11:39:59 +01:00
Andreas Koepf (aider)
6b41e3cc47 fix: Prevent infinite loop in arithmetic dataset space generation 2025-01-23 11:34:18 +01:00
Andreas Koepf (aider)
473a2be31d feat: Unify arithmetic task generation with configurable dataset class
This commit introduces a new `ArithmeticDataset` class that:
- Combines complex and simple arithmetic task generation approaches
- Provides configurable task generation via `ArithmeticDatasetConfig`
- Supports deterministic task generation
- Implements dataset interface for easy use with HuggingFace datasets
- Adds comprehensive test coverage for the new implementation

Changes include:
- Refactored `basic_arithmetic.py` to use a unified dataset approach
- Added configuration validation and flexible generation options
- Created test suite to validate dataset behavior
- Removed file generation methods in favor of in-memory generation
2025-01-23 11:30:09 +01:00
Andreas Koepf
88e438fa68 feat: Add basic arithmetic module to reasoning_gym 2025-01-23 11:30:03 +01:00
Andreas Koepf (aider)
4dc36f79e7 build: Initialize reasoning_gym package structure with packaging and development setup 2025-01-23 10:50:54 +01:00