Skip to content

Commit 90063c7

Browse files
committed
add
1 parent 1129e53 commit 90063c7

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/test_models.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff 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+
698699
def 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+
710712
def 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?"

0 commit comments

Comments
 (0)