From 21f4817444e3b84ab54918e6330b7c7da987c16f Mon Sep 17 00:00:00 2001 From: Sai-Suraj-27 Date: Wed, 24 Jul 2024 22:17:11 +0530 Subject: [PATCH] Replaced deprecated unittest method with the correct one. --- tests/models/llava/test_processor_llava.py | 2 +- tests/models/llava_next/test_processor_llava_next.py | 2 +- tests/models/vipllava/test_processor_vipllava.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) 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)