Fix Bloom test

Rocketknight1 · Rocketknight1 · commit 4942e98ddd4c · 2024-07-22T16:07:59.000+01:00
diff --git a/tests/models/bloom/test_tokenization_bloom.py b/tests/models/bloom/test_tokenization_bloom.py
@@ -135,7 +135,7 @@ def test_encodings_from_xnli_dataset(self):
     @require_jinja
     def test_tokenization_for_chat(self):
         tokenizer = self.get_rust_tokenizer()
-        tokenizer.chat_template = "{% for message in messages %}{{'<|im_start|>' + message['role'] + '\n' + message['content'] + '<|im_end|>' + '\n'}}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant\n' }}{% endif %}"
+        tokenizer.chat_template = "{% for message in messages %}" "{{ message.content }}{{ eos_token }}" "{% endfor %}"
         test_chats = [
             [{"role": "system", "content": "You are a helpful chatbot."}, {"role": "user", "content": "Hello!"}],
             [

Original file line number	Diff line number	Diff line change
`@@ -135,7 +135,7 @@ def test_encodings_from_xnli_dataset(self):`
`135`	`135`	`@require_jinja`
`136`	`136`	`def test_tokenization_for_chat(self):`
`137`	`137`	`tokenizer = self.get_rust_tokenizer()`
`138`		`- tokenizer.chat_template = "{% for message in messages %}{{'<\|im_start\|>' + message['role'] + '\n' + message['content'] + '<\|im_end\|>' + '\n'}}{% endfor %}{% if add_generation_prompt %}{{ '<\|im_start\|>assistant\n' }}{% endif %}"`
	`138`	`+ tokenizer.chat_template = "{% for message in messages %}" "{{ message.content }}{{ eos_token }}" "{% endfor %}"`
`139`	`139`	`test_chats = [`
`140`	`140`	`[{"role": "system", "content": "You are a helpful chatbot."}, {"role": "user", "content": "Hello!"}],`
`141`	`141`	`[`