fix: Replaced deprecated unittest method with the correct one (#32198)

Replaced deprecated unittest method with the correct one.
2024-07-24 22:30:21 +05:30
parent edd68f4ed8
commit 85a1269e19
3 changed files with 3 additions and 3 deletions
--- a/tests/models/llava/test_processor_llava.py
+++ b/tests/models/llava/test_processor_llava.py
@@ -44,4 +44,4 @@ class LlavaProcessorTest(unittest.TestCase):
        ]

        formatted_prompt = processor.apply_chat_template(messages, add_generation_prompt=True)
-        self.assertEquals(expected_prompt, formatted_prompt)
+        self.assertEqual(expected_prompt, formatted_prompt)