[ci] fix phi3 test

RUCAIBox · Jun 22, 2024 · d5daabc · d5daabc
1 parent 9aa2990
commit d5daabc
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/tests/utilization/model/test_apply_prompt_template.py b/tests/utilization/model/test_apply_prompt_template.py
@@ -57,15 +57,15 @@ def test_phi3(conversation: Conversation):
     formatted_conversation = conversation.apply_prompt_template()
     assert formatted_conversation == (
         "<s><|system|>\n"
-        "This is a system message.<|end|>\n"
+        "This is a system message. <|end|>\n"
         "<|user|>\n"
-        "This is a user message.<|end|>\n"
+        "This is a user message. <|end|>\n"
         "<|assistant|>\n"
-        "This is an assistant message.<|end|>\n"
+        "This is an assistant message. <|end|>\n"
         "<|user|>\n"
-        "This is the second user message.<|end|>\n"
+        "This is the second user message. <|end|>\n"
         "<|assistant|>\n"
-        "This is the second assistant message.<|end|>\n"
+        "This is the second assistant message. <|end|>\n"
     )