Andreas Koepf (aider)
d7ef1d471b
feat: Add GCD dataset imports to arithmetic module
2025-01-24 08:51:29 +01:00
Andreas Koepf (aider)
c88ec95762
feat: Add GCD dataset generator for finding GCD of multiple integers
2025-01-24 08:47:19 +01:00
Andreas Koepf (aider)
3c1232888c
feat: Clarify prime factorization example and answer format
2025-01-23 22:50:00 +01:00
Andreas Koepf (aider)
a4391fe5f6
feat: Add prime factorization task generator with configurable range and example
2025-01-23 22:46:58 +01:00
Andreas Koepf (aider)
562dfb1813
refactor: Rename chain_sum to chain_sum_dataset for consistency
2025-01-23 22:27:48 +01:00
Andreas Koepf (aider)
2b58508701
feat: Update arithmetic __init__.py with docstring and additional exports
2025-01-23 22:26:51 +01:00
Andreas Koepf (aider)
b8c69aa89f
feat: Add leg counting imports and exports to arithmetic package
2025-01-23 22:22:12 +01:00
Andreas Koepf (aider)
f8219a7a6a
refactor: Remove centipede from ANIMALS due to variable leg count
2025-01-23 22:18:59 +01:00
Andreas Koepf (aider)
0a239af02d
refactor: Sort ANIMALS dictionary by leg count with alphabetical order within groups
2025-01-23 22:17:27 +01:00
Andreas Koepf (aider)
8f3a52350c
feat: Add more 4-legged and 6-legged animals to ANIMALS dictionary
2025-01-23 22:16:08 +01:00
Andreas Koepf (aider)
6184941d6c
feat: Add more diverse animals to leg counting dataset
2025-01-23 22:14:15 +01:00
Andreas Koepf (aider)
ce9076d7af
feat: Add leg counting arithmetic task generator with animal leg counting functionality
2025-01-23 22:08:42 +01:00
Andreas Koepf (aider)
85f6f63cb1
feat: Add configurable whitespace option to ArithmeticDataset
2025-01-23 14:25:54 +01:00
Andreas Koepf (aider)
dae0cff2f5
refactor: Reorder parameters of _format_question to match other methods
2025-01-23 14:22:46 +01:00
Andreas Koepf (aider)
dce4c7eba5
refactor: Pass item_rng to _format_question() method in ArithmeticDataset
2025-01-23 14:20:22 +01:00
Andreas Koepf (aider)
8d1dac9e62
feat: Add arithmetic_dataset() factory function to basic_arithmetic.py
2025-01-23 12:47:01 +01:00
Andreas Koepf
0aa35e15a3
style: Format code with consistent whitespace and remove unnecessary lines
2025-01-23 12:45:15 +01:00
Andreas Koepf (aider)
ba493adbe7
refactor: Remove duplicate chain_sum function from basic_arithmetic.py
2025-01-23 12:41:18 +01:00
Andreas Koepf (aider)
d3cc4b0d0e
feat: Add chain_sum() factory function to create ChainSum dataset instances
2025-01-23 12:40:50 +01:00
Andreas Koepf (aider)
f42d6c2212
fix: Ensure deterministic iteration for ArithmeticDataset
2025-01-23 12:37:40 +01:00
Andreas Koepf (aider)
05ba566a51
feat: Add iteration support to ArithmeticDataset with size-based termination
2025-01-23 12:34:54 +01:00
Andreas Koepf (aider)
575624aa93
feat: Expose arithmetic modules automatically in reasoning_gym package
2025-01-23 12:25:26 +01:00
Andreas Koepf (aider)
5c5cbf3194
feat: Add iterator support to ChainSum with size-respecting iteration
2025-01-23 12:23:35 +01:00
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
