Commit graph

32 commits

Author SHA1 Message Date
Andreas Koepf
476e37e70b use Decimal class for numeric comparison e.g. +0123.100 == 123.1 2025-02-21 15:36:06 +01:00
Andreas Koepf
ff5b210106 use native types List->list, Dict->dict, Set->set, Tuple->tuple 2025-02-21 15:15:38 +01:00
AhmedSaif2
6b5c7a8637 add a helper function to handle redundant code 2025-02-21 15:54:00 +02:00
AhmedSaif2
27c597c10a remove added instructions 2025-02-20 21:29:01 +02:00
AhmedSaif2
1da1909666 Add score answer to support comma format 2025-02-20 20:52:31 +02:00
Andreas Koepf
aa794253fe minor formatting changes 2025-02-17 18:20:18 +01:00
abdulhakeem
5b1e42f878 Tweaked some question templates 2025-02-17 02:58:42 -06:00
Andreas Koepf
6bf2dfa36c formatting 2025-02-16 16:18:39 +01:00
joesharratt1229
569517664f corrected failing airthmetic test 2025-02-16 12:01:54 +00:00
joesharratt1229
a59e4cc918 reformatted prompt 2025-02-16 09:27:21 +00:00
joesharratt1229
c28688cb96 reformatted basic airth question template 2025-02-16 09:07:56 +00:00
Andreas Koepf
1996ffa6d8 add ProductsDataset (multiplication tasks) 2025-02-13 17:59:02 +01:00
Andreas Koepf
519e411fa5 add reasoning_gym.create_dataset({name}, ...) global factory function 2025-01-25 00:58:34 +01:00
Andreas Koepf
0d2d8ba6a0 pass config to ProceduralDataset base 2025-01-25 00:23:05 +01:00
Andreas Koepf
9d436af2b9 Add support for integer division in basic arithmetic dataset 2025-01-24 15:07:31 +01:00
Andreas Koepf (aider)
3ac07f6558 feat: Add support for integer division in BasicArithmeticDataset 2025-01-24 11:30:38 +01:00
Andreas Koepf
aaabc05ace formatting 2025-01-24 10:34:07 +01:00
Andreas Koepf
0e9250bce0 Rename ArithmeticDataset to BasicArithmeticDataset 2025-01-24 10:31:26 +01:00
Andreas Koepf (aider)
e4b032c814 refactor: Rename ArithmeticDataset to BasicArithmeticDataset 2025-01-24 10:02:02 +01:00
Andreas Koepf (aider)
339d515ecf refactor: Inherit ArithmeticDataset from ProceduralDataset base class 2025-01-24 10:00:19 +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)
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