Skip to content

Commit 8e0e0ef

Browse files
committed
fix: keep polling running rollout statuses
Continue polling when the lightweight status endpoint returns RUNNING so the remote rollout processor only exits the poll loop for terminal statuses. Made-with: Cursor
1 parent 20b0f23 commit 8e0e0ef

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

eval_protocol/pytest/remote_rollout_processor.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -129,6 +129,9 @@ async def _process_row(row: EvaluationRow) -> EvaluationRow:
129129
status = (status_result or {}).get("status")
130130
if isinstance(status, dict) and "code" in status:
131131
status_code = status["code"]
132+
if status_code == Status.Code.RUNNING:
133+
await asyncio.sleep(poll_interval)
134+
continue
132135

133136
logger.info(
134137
"Found status for rollout %s with code %s",

0 commit comments

Comments
 (0)