From 2c1eef34cbd5c1d3ccc3c1421b14dc875a659a6c Mon Sep 17 00:00:00 2001
From: hiyouga <467089858@qq.com>
Date: Wed, 4 Sep 2024 22:38:26 +0800
Subject: [PATCH] fix test

Former-commit-id: 553a83aff9f9da35c9a0eca81f7d2b0bf2adf6ff
---
 tests/data/test_template.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/data/test_template.py b/tests/data/test_template.py
index 0c5e669e..b5cf0f82 100644
--- a/tests/data/test_template.py
+++ b/tests/data/test_template.py
@@ -117,7 +117,7 @@ def test_encode_multiturn(use_fast: bool):
 def test_jinja_template(use_fast: bool):
     tokenizer = AutoTokenizer.from_pretrained(TINY_LLAMA, use_fast=use_fast)
     ref_tokenizer = AutoTokenizer.from_pretrained(TINY_LLAMA, use_fast=use_fast)
-    get_template_and_fix_tokenizer(tokenizer, name="llama3")
+    get_template_and_fix_tokenizer(tokenizer, DataArguments(template="llama3"))
     assert tokenizer.chat_template != ref_tokenizer.chat_template
     assert tokenizer.apply_chat_template(MESSAGES) == ref_tokenizer.apply_chat_template(MESSAGES)