agent-lightning icon indicating copy to clipboard operation
agent-lightning copied to clipboard

Add batch rollout helpers and multi-task queueing

Open ultmaster opened this issue 4 months ago • 0 comments

Summary

  • allow training_rollout_batch/validation_rollout_batch APIs to receive per-task resource lists
  • add batch_size to AgentRunner and propagate through Trainer for 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.py
  • pytest tests/test_litagent.py tests/test_client.py

https://chatgpt.com/codex/tasks/task_e_689db4c153ec832ea2983ab0885c7f75

ultmaster avatar Aug 15 '25 02:08 ultmaster