Skip to content

Commit fb8debe

Browse files
committed
remove print
1 parent c726a57 commit fb8debe

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

eval_protocol/pytest/default_single_turn_rollout_process.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -97,10 +97,7 @@ async def process_row(row: EvaluationRow) -> EvaluationRow:
9797
chunks.append(chunk)
9898
response = litellm.stream_chunk_builder(chunks, messages_payload)
9999
else:
100-
tc = time.perf_counter()
101-
# print(f"run_id {row.execution_metadata.run_id} request_params: {json.dumps(request_params)}")
102100
response = await acompletion(**request_params)
103-
print(f"run_id {row.execution_metadata.run_id} time taken: {time.perf_counter() - tc} speculation_enabled: {request_params.get('extra_body', {}).get('prediction', None) is not None}")
104101

105102
assert response is not None, "Response is None"
106103
assert isinstance(response, ModelResponse), "Response should be ModelResponse"

0 commit comments

Comments
 (0)