Skip to content

Commit 5488bfe

Browse files
quick fix
1 parent 2061e71 commit 5488bfe

1 file changed

Lines changed: 7 additions & 2 deletions

File tree

eval_protocol/pytest/remote_rollout_processor.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -125,8 +125,13 @@ async def _process_row(row: EvaluationRow) -> EvaluationRow:
125125
status_result = await asyncio.to_thread(
126126
self._tracing_adapter.get_status, rollout_id=row.execution_metadata.rollout_id
127127
)
128-
if status_result and status_result.get("status"):
129-
status_code = status_result["status"]["code"]
128+
status = (status_result or {}).get("status")
129+
if status and "code" in status:
130+
status_code = status["code"]
131+
132+
if status_code == Status.Code.RUNNING:
133+
await asyncio.sleep(poll_interval)
134+
continue
130135

131136
logger.info(
132137
"Found status for rollout %s with code %s",

0 commit comments

Comments
 (0)