File tree Expand file tree Collapse file tree 1 file changed +3
-1
lines changed
Expand file tree Collapse file tree 1 file changed +3
-1
lines changed Original file line number Diff line number Diff line change @@ -695,6 +695,7 @@ def test_evaluation_row_extra_fields():
695695 assert "accuracy" in dictionary ["eval_details" ]["metrics" ]
696696 assert "test" in dictionary ["extra_fields" ]
697697
698+
698699def test_message_with_weight_dump ():
699700 example = {
700701 "role" : "user" ,
@@ -707,6 +708,7 @@ def test_message_with_weight_dump():
707708 assert "weight" in dictionary
708709 assert dictionary ["weight" ] == 0
709710
711+
710712def test_message_dump_for_chat_completion_request ():
711713 example = {
712714 "role" : "user" ,
@@ -718,4 +720,4 @@ def test_message_dump_for_chat_completion_request():
718720 dictionary = message .dump_mdoel_for_chat_completion_request ()
719721 assert "weight" not in dictionary
720722 assert "reasoning_content" not in dictionary
721- assert dictionary ["content" ] == "Hello, how are you?"
723+ assert dictionary ["content" ] == "Hello, how are you?"
You can’t perform that action at this time.
0 commit comments