Skip to content

Commit 462a572

Browse files
committed
fix
1 parent 6d9724f commit 462a572

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

eval_protocol/pytest/remote_rollout_processor.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -132,8 +132,7 @@ async def _process_row(row: EvaluationRow) -> EvaluationRow:
132132
# Use the first log with status information
133133
status_log = status_logs[0]
134134
status_dict = status_log.get("status")
135-
# Filter out transport metadata fields from extras
136-
raw_extras = status_log.get("extras", {})
135+
raw_extras = status_log.get("extras") or {}
137136
status_extras = {
138137
k: v for k, v in raw_extras.items() if k not in ("logger_name", "level", "timestamp")
139138
}
@@ -161,7 +160,10 @@ async def _process_row(row: EvaluationRow) -> EvaluationRow:
161160
details=status_details,
162161
)
163162

164-
row.execution_metadata.extra = status_extras
163+
if row.execution_metadata.extra:
164+
row.execution_metadata.extra.update(status_extras)
165+
else:
166+
row.execution_metadata.extra = status_extras
165167

166168
logger.info("Stopping polling for rollout %s", row.execution_metadata.rollout_id)
167169
break

0 commit comments

Comments
 (0)