diff --git a/tests/models/llava/test_processor_llava.py b/tests/models/llava/test_processor_llava.py index b668b6f4d6d843..54c1b4674cbcef 100644 --- a/tests/models/llava/test_processor_llava.py +++ b/tests/models/llava/test_processor_llava.py @@ -44,4 +44,4 @@ def test_chat_template(self): ] formatted_prompt = processor.apply_chat_template(messages, add_generation_prompt=True) - self.assertEquals(expected_prompt, formatted_prompt) + self.assertEqual(expected_prompt, formatted_prompt) diff --git a/tests/models/llava_next/test_processor_llava_next.py b/tests/models/llava_next/test_processor_llava_next.py index 0a6eccf555e7b0..0d2deacdabaa9f 100644 --- a/tests/models/llava_next/test_processor_llava_next.py +++ b/tests/models/llava_next/test_processor_llava_next.py @@ -38,4 +38,4 @@ def test_chat_template(self): ] formatted_prompt = processor.apply_chat_template(messages, add_generation_prompt=True) - self.assertEquals(expected_prompt, formatted_prompt) + self.assertEqual(expected_prompt, formatted_prompt) diff --git a/tests/models/vipllava/test_processor_vipllava.py b/tests/models/vipllava/test_processor_vipllava.py index 896eddd09c7130..0ddf569a60f58d 100644 --- a/tests/models/vipllava/test_processor_vipllava.py +++ b/tests/models/vipllava/test_processor_vipllava.py @@ -38,4 +38,4 @@ def test_chat_template(self): ] formatted_prompt = processor.apply_chat_template(messages, add_generation_prompt=True) - self.assertEquals(expected_prompt, formatted_prompt) + self.assertEqual(expected_prompt, formatted_prompt)