Skip to content

Commit a71074e

Browse files
committed
update remote rollout processor
1 parent d569a64 commit a71074e

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

eval_protocol/pytest/remote_rollout_processor.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,7 @@ async def _process_row(row: EvaluationRow) -> EvaluationRow:
9494
def _post_init() -> None:
9595
url = f"{remote_base_url}/init"
9696
try:
97-
r = requests.post(url, json=init_payload.model_dump(), timeout=30)
97+
r = requests.post(url, json=init_payload.model_dump(), timeout=300)
9898
r.raise_for_status()
9999
except requests.exceptions.Timeout:
100100
raise TimeoutError(
@@ -133,9 +133,9 @@ def _get_status() -> Dict[str, Any]:
133133
# For all other exceptions, raise them
134134
raise
135135

136-
# Search Fireworks tracing logs for completion
137-
completed_logs = self._tracing_adapter.search_logs(
138-
tags=[f"rollout_id:{row.execution_metadata.rollout_id}"]
136+
# Search Fireworks tracing logs for completion (run in thread to avoid blocking event loop)
137+
completed_logs = await asyncio.to_thread(
138+
self._tracing_adapter.search_logs, tags=[f"rollout_id:{row.execution_metadata.rollout_id}"]
139139
)
140140
# Filter for logs that actually have status information
141141
status_logs = []

0 commit comments

Comments
 (0)