Skip to content

Commit 6b4db20

Browse files
author
Dylan Huang
committed
fix types
1 parent 4d63471 commit 6b4db20

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

eval_protocol/pytest/default_pydantic_ai_rollout_processor.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -103,7 +103,7 @@ async def _sem_wrapper(r: EvaluationRow) -> EvaluationRow:
103103

104104
async def convert_pyd_message_to_ep_message(self, messages: list[ModelMessage]) -> list[Message]:
105105
oai_messages: list[ChatCompletionMessageParam] = await self.util._map_messages(messages)
106-
return [Message(role=m["role"], **m) for m in oai_messages] # pyright: ignore[reportArgumentType]
106+
return [Message(**m) for m in oai_messages] # pyright: ignore[reportArgumentType]
107107

108108
def convert_ep_message_to_pyd_message(self, message: Message, row: EvaluationRow) -> ModelMessage:
109109
if message.role == "assistant":

0 commit comments

Comments
 (0)