Skip to content

Commit e5ef516

Browse files
quick fix
1 parent d162501 commit e5ef516

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
@@ -124,8 +124,13 @@ async def _process_row(row: EvaluationRow) -> EvaluationRow:
124124
status_result = await asyncio.to_thread(
125125
self._tracing_adapter.get_status, rollout_id=row.execution_metadata.rollout_id
126126
)
127-
if status_result and status_result.get("status"):
128-
status_code = status_result["status"]["code"]
127+
status = (status_result or {}).get("status")
128+
if status and "code" in status:
129+
status_code = status["code"]
130+
131+
if status_code == Status.Code.RUNNING:
132+
await asyncio.sleep(poll_interval)
133+
continue
129134

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

0 commit comments

Comments
 (0)