We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 2061e71 commit 5488bfeCopy full SHA for 5488bfe
1 file changed
eval_protocol/pytest/remote_rollout_processor.py
@@ -125,8 +125,13 @@ async def _process_row(row: EvaluationRow) -> EvaluationRow:
125
status_result = await asyncio.to_thread(
126
self._tracing_adapter.get_status, rollout_id=row.execution_metadata.rollout_id
127
)
128
- if status_result and status_result.get("status"):
129
- status_code = status_result["status"]["code"]
+ status = (status_result or {}).get("status")
+ 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
135
136
logger.info(
137
"Found status for rollout %s with code %s",
0 commit comments