Commit graph

837 commits

Author SHA1 Message Date
shannonsands
46f0602227
Diplomacy trainer env (#227)
* minimal implementation, simplified challenge registry

* need game save logic

* fixed challenge gen, works with local test

* updated challenge gen with wider ranges, working with local script

* runs working correctly, wandb stats look ok

* linting

* Add diplomacy environment with AI_Diplomacy submodule

- Add diplomacy_env_minimal.py for diplomacy game environment
- Add atropos_client_minimal.py for client interface
- Add diplomacy_local_server.py for local game server
- Add AI_Diplomacy submodule from GoodStartLabs/AI_Diplomacy
- Fix import ordering and remove unused imports

* test file working, moving to cluster to test training

* updated gitignore

* removed logs

* minor fixes, training running now

* readded proxy reg and queue system

* linting

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* queue gameid bug, refactored

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* cleaned up configs & allowed for openrouter models to be easily used

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* linting

* Remove duplicate dependencies from diplomacy requirements.txt

Only keep AI_Diplomacy-specific dependencies that aren't already in the main project

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2025-08-12 09:02:16 +10:00
dmahan93
4fe67e698d
Merge pull request #228 from NousResearch/pre-commit-ci-update-config
[pre-commit.ci] pre-commit autoupdate
2025-08-04 12:31:35 -05:00
pre-commit-ci[bot]
70043c4eb2
[pre-commit.ci] pre-commit autoupdate
updates:
- [github.com/astral-sh/ruff-pre-commit: v0.12.5 → v0.12.7](https://github.com/astral-sh/ruff-pre-commit/compare/v0.12.5...v0.12.7)
2025-08-04 16:41:05 +00:00
shannonsands
47cb15745c
Textworld minimal (#225)
* minimal implementation, simplified challenge registry

* need game save logic

* fixed challenge gen, works with local test

* updated challenge gen with wider ranges, working with local script

* runs working correctly, wandb stats look ok

* linting

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

* removed unused imports

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
2025-08-01 10:16:35 +10:00
Teknium
1900a577d7
Update README.md
Remove hackathon event teaser
2025-07-30 13:56:11 -07:00
Teknium
be66e120d9
Merge pull request #219 from NousResearch/add-arenahard-v1-environment
Add arena-hard v1 environment
2025-07-30 09:35:14 -07:00
pre-commit-ci[bot]
65aea8bb21 [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
2025-07-30 15:10:36 +00:00
teknium
75f1cf6d2a move eval envs to eval_environments and update readmes 2025-07-30 15:09:34 +00:00
shannonsands
9f23c732dd
qwen tokenizer wrapper & fixed jinja template for tool handling (#224)
* added qwen tokenizer wrapper & fixed jinja template for tool handling issues in the official HF one

* moved jinja template into it's own file
2025-07-30 11:57:15 +10:00
dmahan93
56fb50a503
Merge pull request #222 from NousResearch/pre-commit-ci-update-config
[pre-commit.ci] pre-commit autoupdate
2025-07-29 15:52:20 -05:00
dmahan93
9734a10290
Merge pull request #220 from Aboozle1/add-my-environment
Add Word Hunt environment
2025-07-28 12:02:01 -05:00
Aboozle1
3ce68aed38
Merge branch 'main' into add-my-environment 2025-07-28 11:50:50 -05:00
Abhaykhanna3
9d7bcc523f Fix(PR): Address reviewer feedback
- Remove redundant requirements.txt
- Fix leading newline in prompt templates
2025-07-28 11:48:02 -05:00
pre-commit-ci[bot]
d65f7f842d
[pre-commit.ci] pre-commit autoupdate
updates:
- [github.com/astral-sh/ruff-pre-commit: v0.12.4 → v0.12.5](https://github.com/astral-sh/ruff-pre-commit/compare/v0.12.4...v0.12.5)
2025-07-28 16:38:41 +00:00
Abhaykhanna3
b5234d4214 Add Word Hunt environment for training models on 4x4 letter grids
- Trie-based solver, official scoring, normalized rewards
- Configurable token limit and detailed README with dictionary download link
- Removes large Dictionary.txt from tracking and adds ignore rules
- All tests pass and pre-commit hooks are clean
2025-07-28 00:37:36 -05:00
pre-commit-ci[bot]
4c88a4bbb9 [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
2025-07-28 01:37:03 +00:00
teknium
aaebb8d6bb linter linter 2025-07-28 01:35:49 +00:00
pre-commit-ci[bot]
cbec584202 [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
2025-07-28 01:30:27 +00:00
teknium
1d523472cc Merge branch 'add-arenahard-v1-environment' of https://github.com/NousResearch/atropos into add-arenahard-v1-environment 2025-07-28 01:29:52 +00:00
teknium
20565d8abc update judge confs so it can use any judge model 2025-07-28 01:29:50 +00:00
pre-commit-ci[bot]
041a70d891 [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
2025-07-27 03:39:42 +00:00
teknium
e6de7bb432 lint 2025-07-27 03:39:05 +00:00
pre-commit-ci[bot]
52b505296c [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
2025-07-27 02:52:39 +00:00
teknium
a0979eb08e add readme section 2025-07-27 02:46:51 +00:00
teknium
31b0c6f66d Add arena-hard v1 environment 2025-07-26 21:17:00 +00:00
Teknium
b272b7bae9
Merge pull request #218 from NousResearch/pairwise-judgement-env-updates
Pairwise Judgement Environment - improve dataloading, ctx len
2025-07-26 14:14:24 -07:00
pre-commit-ci[bot]
65682d160a [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
2025-07-26 21:13:05 +00:00
teknium
aa66b09c13 make linter happy 2025-07-26 21:12:30 +00:00
pre-commit-ci[bot]
a2e14cf50c [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
2025-07-26 21:08:01 +00:00
teknium
1039c3d360 improve dataloading, ctx len 2025-07-26 21:06:45 +00:00
dmahan93
6604a2255b
Merge pull request #195 from interstellarninja/feat/interleaved_tool_use
Interleaved Tool-Use Within Reasoning Blocks
2025-07-24 08:58:00 -05:00
pre-commit-ci[bot]
97ac993d07 [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
2025-07-24 12:40:01 +00:00
interstellarninja
973864a7e8 resolving further 2025-07-24 08:39:20 -04:00
interstellarninja
09a6f174a8 resolve conflicts and apply hook auto-fixes 2025-07-24 08:23:17 -04:00
dmahan93
1f8183f5d4
Merge pull request #215 from NousResearch/pre-commit-ci-update-config
[pre-commit.ci] pre-commit autoupdate
2025-07-21 12:27:01 -05:00
pre-commit-ci[bot]
a659fbf593
[pre-commit.ci] pre-commit autoupdate
updates:
- [github.com/astral-sh/ruff-pre-commit: v0.12.3 → v0.12.4](https://github.com/astral-sh/ruff-pre-commit/compare/v0.12.3...v0.12.4)
2025-07-21 16:39:06 +00:00
dmahan93
708d24d54b
Update CONFIG.md to fix | in type breaking table 2025-07-21 09:25:34 -05:00
dmahan93
8284a0b95c
Update pyproject.toml 2025-07-16 15:38:16 -05:00
dmahan93
b65bc1b4ba
version update 2025-07-16 15:38:04 -05:00
Teknium
62cee8ac66
Merge pull request #209 from NousResearch/add-pairwise-judge-environment
Add LLM as a judge environment for eval and train based on RewardBench
2025-07-16 13:37:09 -07:00
pre-commit-ci[bot]
6455c305e6 [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
2025-07-16 17:51:15 +00:00
teknium
542185bbcc Merge branch 'add-pairwise-judge-environment' of https://github.com/NousResearch/atropos into add-pairwise-judge-environment 2025-07-16 17:48:44 +00:00
teknium
a43520e619 one last linter... 2025-07-16 17:48:43 +00:00
pre-commit-ci[bot]
eab2c938ea [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
2025-07-16 16:58:42 +00:00
teknium
18f228615d linter stuff 2025-07-16 16:57:51 +00:00
pre-commit-ci[bot]
ffc210e470 [pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
2025-07-16 16:51:19 +00:00
teknium
2f37714e84 Merge branch 'add-pairwise-judge-environment' of https://github.com/NousResearch/atropos into add-pairwise-judge-environment 2025-07-16 16:50:04 +00:00
teknium
0113dc906b add a bunch of extra debugging traces - configurable 2025-07-16 16:49:42 +00:00
Teknium
145762f64a
Merge pull request #212 from sky-coderay/main
fix: correct quantum environment repository URL
2025-07-16 02:45:34 -07:00
Skylar Ray
e889324171
fix: correct quantum environment repository URL 2025-07-16 11:00:45 +03:00