fix(orchestrator): drain after final train batch#5

jooooonas wants to merge 2 commits into

mainp-doom/prime-rl:mainfrom

fix/final-batch-drainp-doom/prime-rl:fix/final-batch-drainCopy head branch name to clipboard