File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed
Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff 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 = []
You can’t perform that action at this time.
0 commit comments