Andreas Koepf (aider)
7a64273f2e
feat: Add support for integer division in BasicArithmeticDataset
2025-01-24 11:30:38 +01:00
Andreas Koepf
20069b2a7d
formatting
2025-01-24 10:34:07 +01:00
Andreas Koepf
98988c8481
Rename ArithmeticDataset to BasicArithmeticDataset
2025-01-24 10:31:26 +01:00
Andreas Koepf (aider)
95d9974911
refactor: Rename ArithmeticDataset to BasicArithmeticDataset
2025-01-24 10:02:02 +01:00
Andreas Koepf (aider)
1b556b70f7
refactor: Inherit ArithmeticDataset from ProceduralDataset base class
2025-01-24 10:00:19 +01:00
Andreas Koepf (aider)
03a3bcf059
feat: Add configurable whitespace option to ArithmeticDataset
2025-01-23 14:25:54 +01:00
Andreas Koepf (aider)
6dbb5b39ee
refactor: Reorder parameters of _format_question to match other methods
2025-01-23 14:22:46 +01:00
Andreas Koepf (aider)
f12e3788d1
refactor: Pass item_rng to _format_question() method in ArithmeticDataset
2025-01-23 14:20:22 +01:00
Andreas Koepf (aider)
48492c4fd8
feat: Add arithmetic_dataset() factory function to basic_arithmetic.py
2025-01-23 12:47:01 +01:00
Andreas Koepf
72ada57dc5
style: Format code with consistent whitespace and remove unnecessary lines
2025-01-23 12:45:15 +01:00
Andreas Koepf (aider)
38801a7e6f
refactor: Remove duplicate chain_sum function from basic_arithmetic.py
2025-01-23 12:41:18 +01:00
Andreas Koepf (aider)
c253b1adbb
feat: Add chain_sum() factory function to create ChainSum dataset instances
2025-01-23 12:40:50 +01:00
Andreas Koepf (aider)
272ebc5a42
fix: Ensure deterministic iteration for ArithmeticDataset
2025-01-23 12:37:40 +01:00
Andreas Koepf (aider)
a8b3a86831
feat: Add iteration support to ArithmeticDataset with size-based termination
2025-01-23 12:34:54 +01:00
Andreas Koepf (aider)
fbba398c91
fix: Prevent infinite loop in arithmetic dataset space generation
2025-01-23 11:34:18 +01:00
Andreas Koepf (aider)
40596262e1
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
8a6364a791
feat: Add basic arithmetic module to reasoning_gym
2025-01-23 11:30:03 +01:00
