927 Commits

Author SHA1 Message Date
Zafir Stojanovski
2c4e45d9a9 Update spiral_matrix.py (#511)
* Improve spiral matrix instructions with clearer movement description and hint

---------

Co-authored-by: Andreas Koepf <andreas.koepf@provisio.com>
2025-10-06 13:02:32 +02:00
Zafir Stojanovski
2f9eaee32a fix: Register missing coin_flip (#507)
* register missing coin_flip

* lint
2025-09-15 14:23:30 +02:00
Kumar Anant
b0815043a2 Add probability dataset (initial: Coin Flip dataset + curriculum) (#505) 2025-09-06 15:59:23 +01:00
Denini Gabriel
02b7fac863 fix encoding to be able to run on win (#502) 2025-08-18 09:19:45 +01:00
Zafir Stojanovski
a969d8ef05 feat(curriculum): Knights and Knaves configs (#488)
* configs

* reduce complexity of curriculum

* update lower bound

* add failure threshold

* update last_k

* update thresholds for success and failure

* update curriculum file as well

* update run name for noncurriculum

* lint

* dtype model eval

* return binary scoring

* set eval repeats to 3

* fix tests
2025-07-31 10:18:05 +02:00
joesharratt1229
4b60c32978 Curr exp (#487)
* began curr exp

* added holdout words

* updated config

* added context

* updated base curriculum

* updaed

* updated curriculum

* updated

* updated

* updated automatic flag

* updated ray trainer

* update
2025-07-25 20:38:47 +01:00
theblackcat102
2d19f13e0f [fix #484] resolve basic_arithmetic fails when size is large (#485)
* [fix] resolve basic_arithmetic fails when size is large by replacing zero divisor with 1
2025-07-07 09:46:23 +01:00
joesharratt1229
876e0aa440 corrected countdown issue (#479) 2025-06-25 13:37:04 -07:00
Adefioye
9e79fc84b6 fix: Rounding issues in score_answer and add unit tests (#462) 2025-06-09 19:18:11 +01:00
Oliver Stanley
5726034a26 fix color_cubes answer strings, update gallery with latest envs (#464)
* update gallery with latest envs

* fix regression where answer str is wrong in color_cubes

* re-update gallery
2025-06-08 13:16:54 +02:00
Oliver Stanley
602e4be0a2 add survo env (#461)
* add survo env

* add survo curriculum

* add survo tests
2025-06-08 11:56:33 +01:00
Oliver Stanley
c2fdb11980 add kakurasu env (#460)
* add kakurasu env

* add kakurasu curriculum

* add kakurasu tests
2025-06-08 09:20:53 +01:00
Oliver Stanley
1a727ecf4e support python 3.10 (#450)
* support python 3.10

* add 3.10 to tests

* new StrEnum
2025-06-04 10:34:01 +01:00
Adefioye
9053009dbe Fix bug in normalize_answer method (#444) 2025-06-02 08:58:54 +02:00
Oliver Stanley
c0e98f93b4 make task entries json serializable (#443)
* make sympy-based task entries json serializable

* remove datetime objs from time_intervals metadata

* make adv geometry json serializable

* make futoshiki metadata json serializable

* fixes

* futoshiki tweaks

* fix adv geometry

* deal with fractions in str representations

* fix

* restore start_time, end_time as str
2025-06-02 08:57:15 +02:00
Zafir Stojanovski
6d9e14729a fix rubiks cube prompt (#433) 2025-05-15 21:31:21 +01:00
Oliver Stanley
3df26e0fb2 fix decimal arithmetic curriculum to respect constraints (#430)
* fix decimal arithmetic curriculum to respect constraints

* update test accordingly
2025-05-06 19:23:19 +01:00
rasdani
72e45e9401 Feat: expose score_answer function without needing to instantiate a dataset (#422)
* feat: get `score_answer` for a given dataset

* fix: `self` error

* add test
2025-04-18 10:36:44 +02:00
joesharratt1229
d0ef136d5b Feat/intragen experiments (#414)
* added curriculum

* readapted readme

* corrected small errors

* Delete eval/eval/r1/algorithmic/word_sorting.json

* removed redundant argument

* added spell

* removed duplicated fit

* changed config

* added composite changes

* added composite changes

* updated yaml

* added spell backward

* updated read me

* added qwen2.5

* added

* Add files via upload

* updated missing trainer func

* updated curr

* updated spell back

* updated correctness score func

* updated configs

* added local evals

* added updates

* updated datasets

* added fsdp to hf utility

* added algorithmic qwen 3b yaml

* updated read me

* updated configs

* added preappend token

* updated with thinking token

* updated test score board

* resolved comments

* added evaluation scripts

* removed results from pr

* added config

* added partial reward scoring

* added evaluation composites

* added training configs

* added games eval

* added rubriks cube

* resolved merge cinflicts

* added games config

* added latest eval configs

* updated strucutre

* Delete training/evaluations/eval_graphs_composite.yaml

---------

Co-authored-by: joesharratt1229 <joesharrat1229@gmail.com>
2025-04-16 08:04:52 +02:00
Oliver Stanley
224532f12a first inter-domain generalisation experiments (#412)
* tweak len reward

* first inter-generalisation experiment config

* update inter algorithmic config

* default to empty config

* fix typo

* change config to match experiment script

* long prompt fixes

* algorithmic training config tweaks

* imports

* update algorithmic training cfgs

* first logic composite config

* fix dset name

* tweaks

* fix syllogisms dataset

* rm temp print

* initial algebra config

* algebra cfg tweaks

* add gc

* add initial games cfg

* rename games cfg

* fix dset name

* fix sokoban metadata

* remove boxnet

* games cfg tweak
2025-04-14 21:06:40 +01:00
Oliver Stanley
ff5407f766 fix boxnet scoring function (#420)
* avoid error print on invalid move format in tower of hanoi

* avoid boxnet error on hallucinated starting posittions
2025-04-14 20:44:24 +01:00
Zafir Stojanovski
71787c6a0e fix(env): Remove custom score function in bf (#418)
* remove custom score function in bf

* pre commit
2025-04-14 11:26:18 +02:00
Zafir Stojanovski
290bfc4fdd (evals): Medium configs (#415)
* updated medium configs

* fix problematic curriculum values / small issues causing exceptions to be raised

* optimus alpha config

* all configs so far

* fix tests
2025-04-14 08:25:31 +02:00
Zafir Stojanovski
dced3bfc45 fix(curriculum): Make boundaries in curriculum more sensible (#407)
* init

* fix tests

* unify codeio

* filtered for libraries not present in reasoning-gym

* fix more bounds

* puzzle24

* knight swap curriculum

* fix number sorting

* fix attributes

* add validation of config in creation of dataset

* dry run for instantiating and validating the datasets

* remove unused imports

* fix curriculum tests to reference newly updated attribute names
2025-04-04 20:24:14 +02:00
Andreas Köpf
7853263650 bump dev version to 0.1.19 2025-04-03 22:19:07 +01:00
Zafir Stojanovski
dafdee621e fix(env): Unify CodeIO datasets (#405)
* unify codeio

* filtered for libraries not present in reasoning-gym
2025-04-02 22:40:03 +02:00
Adefioye
5b653b346c Data collisions notebooks and data (#406)
* Add collisions data

* Fix logic issues in basic_arithmetic and gsm_symbolic data
2025-04-02 09:36:09 +02:00
joesharratt1229
43c739cb3e Feat/curr adj (#394) 2025-04-02 06:39:14 +01:00
Roman Machacek
2c52f33c3a CodeIO HQ Dataset (#382)
* ADD: CodeIO high quality dataset

Based on the dataset for CodeI/O. Annotated using Qwen-Coder and filtered based on the various metrics resulting in high quality filtered dataset, where approx 50% of the original data is kept.

* ADD: Compressed version

* Delete pure json version
2025-04-01 22:34:33 +02:00
Oliver Stanley
415bcb5ace use constant for dataset name in arc agi (#404) 2025-04-01 22:06:31 +02:00
Adefioye
e3af2dd2bd Add curriculum for arc_agi (#372)
* Add curriculum for arc_agi

* Resolve conflicts

* Remove code smell

* Remove unwanted code
2025-04-01 21:17:52 +02:00
Zafir Stojanovski
50846c3534 fix(env): ARC 1D curriculum (#402)
* Add arc_1d curriculum

* Add difficulty to metadata

* use range attribute instead of scalar

---------

Co-authored-by: abdulhakeem <abdulhakeemadefioye@gmail.com>
Co-authored-by: Oliver Stanley <olivergestanley@gmail.com>
2025-04-01 13:01:15 +02:00
vncntt
cd85c2d632 add knights knaves curriculum (#401)
* add knights knaves curriculum

* add metadata + width constraints
2025-04-01 12:20:58 +02:00
Tiago Serafim
7ae2942c34 Fix small bit of old code (#386)
Removed the `sat_utils.` that was left there from the original zebra package.

This allows the zebra generator to be used with the full set of available clues.
2025-03-24 09:00:28 +00:00
Zafir Stojanovski
2a970390e8 remove coach and update tests (#393) 2025-03-24 00:10:01 +01:00
Zafir Stojanovski
ce0a6c4878 fix(envs): Add source dataset and index to metadata (#388)
* add source dataset and index to metadata

* fix typo

* fix coach class and its test
2025-03-20 11:12:14 +00:00
Oliver Stanley
7475a20700 include ranges rather than sampled values in difficulty metadata dicts (#387)
* update difficulty metadata for logic datasets

* update difficulty metadata for graph datasets

* update difficulty metadata for geometry datasets

* update difficulty metadata for games datasets

* update difficulty metadata for cognition datasets

* update difficulty metadata for arithmetic datasets

* update difficulty metadata for arc datasets

* update difficulty metadata for algorithmic datasets

* update difficulty metadata for algebra datasets

* use tuples

* update tests

* update tests
2025-03-20 10:27:03 +01:00
Rich Jones
b69c35818a fix figlet font curr imports (#383) 2025-03-18 23:38:17 +01:00
Zafir Stojanovski
814da6e08a add difficulty metadata 2025-03-18 13:51:43 +01:00
Zafir Stojanovski
29bf78293f figlet font curriculum 2025-03-18 13:51:43 +01:00
Andreas Koepf
1511c5e301 don't pass answer value to eval 2025-03-17 23:13:53 +01:00
Jean Kaddour
d6aad5a329 fix: add score_answer() to number_sorting (#380)
* fix: add score_answer() to number_sorting

* chore: run pre-commit

* fix: use json.loads()

* fix: run isort()
2025-03-17 23:04:13 +01:00
Andreas Koepf
f25820a0c7 bump version 2025-03-16 15:54:03 +01:00
Andreas Koepf
08b422f673 reduce values in NeedleHaystackCurriculum 2025-03-16 15:52:48 +01:00
Andreas Köpf
d2c895f1d3 Refactor Curriculum Attributes (#335)
* remove min_value from AttributeDefinition
* remove type from AttributeDefinition
* Add CurriculumContext
* add ensure_interval option for RangeAttributes
* docs: Add legend explaining curriculum indicators in dataset gallery
* update GALLERY.md
2025-03-16 15:40:28 +01:00
Andreas Köpf
d6f399b8e4 Add eval configs, small fixes to eval script & rush-hour score_answer 2025-03-16 09:18:05 +01:00
Adefioye
8a0cacc054 Add jugs curriculum (#369) 2025-03-14 18:04:33 +01:00
Andreas Koepf
8c12fe86e2 bump version 2025-03-14 16:21:34 +01:00
Zafir Stojanovski
f72ecaf411 circuit logic curriculum (#368) 2025-03-14 16:17:00 +01:00
Rich Jones
e7d05d6510 GoL-Halt Curricula (#366)
* GoL-Halt Curricula

* trivial
2025-03-14 16:15:45 +01:00