Skip to content

Commit 6702c55

Browse files
committed
update
1 parent b6eac99 commit 6702c55

File tree

2 files changed

+2
-6
lines changed

2 files changed

+2
-6
lines changed

eval_protocol/pytest/github_action_rollout_processor.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -85,12 +85,10 @@ async def _process_row(row: EvaluationRow) -> EvaluationRow:
8585
config.completion_params = (
8686
normalize_fireworks_model_for_litellm(config.completion_params) or config.completion_params
8787
)
88+
row.input_metadata.completion_params = config.completion_params
8889

8990
init_request = build_init_request(row, config, self.model_base_url)
9091

91-
# Update row with normalized completion_params for downstream access
92-
row.input_metadata.completion_params = init_request.completion_params
93-
9492
def _dispatch_workflow():
9593
url = f"https://api.github.com/repos/{self.owner}/{self.repo}/actions/workflows/{self.workflow_id}/dispatches"
9694

eval_protocol/pytest/remote_rollout_processor.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -92,12 +92,10 @@ async def _process_row(row: EvaluationRow) -> EvaluationRow:
9292
config.completion_params = (
9393
normalize_fireworks_model_for_litellm(config.completion_params) or config.completion_params
9494
)
95+
row.input_metadata.completion_params = config.completion_params
9596

9697
init_payload = build_init_request(row, config, model_base_url)
9798

98-
# Update row with normalized completion_params for downstream access
99-
row.input_metadata.completion_params = init_payload.completion_params
100-
10199
# Fire-and-poll
102100
init_url = f"{remote_base_url}/init"
103101

0 commit comments

Comments
 (0)