Skip to content

Commit fc1e273

Browse files
committed
add test
1 parent becb307 commit fc1e273

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/remote_server/test_remote_fireworks.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -141,4 +141,8 @@ async def test_remote_rollout_and_fetch_fireworks(row: EvaluationRow) -> Evaluat
141141
assert "data_loader_type" in row.input_metadata.dataset_info
142142
assert "data_loader_num_rows" in row.input_metadata.dataset_info
143143

144+
assert row.execution_metadata.finish_reason == "stop", (
145+
f"Expected finish_reason='stop', got {row.execution_metadata.finish_reason}"
146+
)
147+
144148
return row

0 commit comments

Comments
 (0)