agent-lightning
agent-lightning copied to clipboard
Add batch rollout helpers and multi-task queueing
Summary
- allow
training_rollout_batch/validation_rollout_batchAPIs to receive per-task resource lists - add
batch_sizetoAgentRunnerand propagate throughTrainerfor batched polling - test batch rollouts with resource lists
Testing
pre-commit run --files agentlightning/litagent.py agentlightning/runner.py agentlightning/trainer.py tests/test_litagent.pypytest tests/test_litagent.py tests/test_client.py
https://chatgpt.com/codex/tasks/task_e_689db4c153ec832ea2983ab0885c7f75