Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
9 changes: 8 additions & 1 deletion eval_protocol/pytest/tracing_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -171,7 +171,14 @@ def update_row_with_remote_trace(
row.messages = remote_row.messages
row.tools = remote_row.tools
row.input_metadata.session_data = remote_row.input_metadata.session_data
row.input_metadata.dataset_info = remote_row.input_metadata.dataset_info
remote_info = remote_row.input_metadata.dataset_info or {}
if row.input_metadata.dataset_info is None:
row.input_metadata.dataset_info = dict(remote_info)
else:
for k, v in remote_info.items():
if k not in row.input_metadata.dataset_info:
row.input_metadata.dataset_info[k] = v

row.execution_metadata = remote_row.execution_metadata
return None
else:
Expand Down
Loading