From 21326e4ef38b6ba35e0fe65eb95281d2b6820628 Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Fri, 11 Oct 2024 11:40:42 +0700 Subject: [PATCH] chore: lint --- tests/prompt_strategies/test_chat_templates.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/tests/prompt_strategies/test_chat_templates.py b/tests/prompt_strategies/test_chat_templates.py index 7a5582ddc..4ec12b82c 100644 --- a/tests/prompt_strategies/test_chat_templates.py +++ b/tests/prompt_strategies/test_chat_templates.py @@ -229,7 +229,9 @@ class TestSharegptChatTemplateLlama3: LOG.info("Testing ShareGPT style datasets with llama-3 assistant prompts") # pylint: disable=duplicate-code strategy = ChatTemplateStrategy( - ChatTemplatePrompter(llama3_tokenizer, chat_template=get_chat_template("llama3")), + ChatTemplatePrompter( + llama3_tokenizer, chat_template=get_chat_template("llama3") + ), tokenizer=llama3_tokenizer, train_on_inputs=False, train_on_eos="none", @@ -280,7 +282,9 @@ class TestSharegptChatTemplateLlama3: LOG.info("Testing ShareGPT style datasets with llama-3 human prompts") # pylint: disable=duplicate-code strategy = ChatTemplateStrategy( - ChatTemplatePrompter(llama3_tokenizer, chat_template=get_chat_template("llama3")), + ChatTemplatePrompter( + llama3_tokenizer, chat_template=get_chat_template("llama3") + ), tokenizer=llama3_tokenizer, train_on_inputs=False, train_on_eos="none", @@ -331,7 +335,9 @@ class TestSharegptChatTemplateLlama3: LOG.info("Testing ShareGPT style datasets with llama-3 system/human prompts") # pylint: disable=duplicate-code strategy = ChatTemplateStrategy( - ChatTemplatePrompter(llama3_tokenizer, chat_template=get_chat_template("llama3")), + ChatTemplatePrompter( + llama3_tokenizer, chat_template=get_chat_template("llama3") + ), tokenizer=llama3_tokenizer, train_on_inputs=False, train_on_eos="none",