fix: test

2024-10-11 11:11:19 +07:00
parent 2038255052
commit e3efa29cf5
1 changed files with 1 additions and 1 deletions
--- a/tests/prompt_strategies/test_chat_templates.py
+++ b/tests/prompt_strategies/test_chat_templates.py
@@ -73,7 +73,7 @@ class TestAssistantChatTemplateLlama3:
        strategy = ChatTemplateStrategy(
            ChatTemplatePrompter(
                llama3_tokenizer,
-                chat_templates("llama3"),
+                chat_template=get_chat_template("llama3"),
                message_field_role="role",
                message_field_content="content",
                roles={