Skip to content

Commit 70b2e17

Browse files
committed
fix
1 parent 01c0c1e commit 70b2e17

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

eval_protocol/pytest/priority_scheduler.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -176,6 +176,14 @@ async def _run_eval(rows_to_eval: Union[EvaluationRow, List[EvaluationRow]]):
176176
processed_row=rows_to_eval,
177177
)
178178
eval_duration = time.perf_counter() - start_time
179+
180+
# Set eval_duration_seconds BEFORE buffer writes to ensure it's included in serialization
181+
if isinstance(eval_res, list):
182+
for row in eval_res:
183+
row.execution_metadata.eval_duration_seconds = eval_duration
184+
else:
185+
eval_res.execution_metadata.eval_duration_seconds = eval_duration
186+
179187
# push result to the output buffer
180188
if self.output_buffer:
181189
if isinstance(eval_res, list):
@@ -188,13 +196,11 @@ async def _run_eval(rows_to_eval: Union[EvaluationRow, List[EvaluationRow]]):
188196

189197
if isinstance(eval_res, list):
190198
for row in eval_res:
191-
row.execution_metadata.eval_duration_seconds = eval_duration
192199
self.results.append(row)
193200
# Update eval progress bar (groupwise: 1 eval for the group)
194201
if self.eval_pbar:
195202
self.eval_pbar.update(1)
196203
else:
197-
eval_res.execution_metadata.eval_duration_seconds = eval_duration
198204
self.results.append(eval_res)
199205
# Update eval progress bar (pointwise: 1 eval per row)
200206
if self.eval_pbar:

0 commit comments

Comments
 (0)