Skip to content

GitLab

Unverified Commit 85a1269e authored by Sai-Suraj-27's avatar Sai-Suraj-27 Committed by GitHub
Browse files

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

Replaced deprecated unittest method with the correct one.
parent edd68f4e
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
tests/models/llava/test_processor_llava.py
View file @ 85a1269e
...@@ -44,4 +44,4 @@ class LlavaProcessorTest(unittest.TestCase): ...@@ -44,4 +44,4 @@ class LlavaProcessorTest(unittest.TestCase):
] ]
formatted_prompt = processor.apply_chat_template(messages, add_generation_prompt=True) formatted_prompt = processor.apply_chat_template(messages, add_generation_prompt=True)
self.assertEquals(expected_prompt, formatted_prompt) self.assertEqual(expected_prompt, formatted_prompt)
tests/models/llava_next/test_processor_llava_next.py
View file @ 85a1269e
...@@ -38,4 +38,4 @@ class LlavaProcessorTest(unittest.TestCase): ...@@ -38,4 +38,4 @@ class LlavaProcessorTest(unittest.TestCase):
] ]
formatted_prompt = processor.apply_chat_template(messages, add_generation_prompt=True) formatted_prompt = processor.apply_chat_template(messages, add_generation_prompt=True)
self.assertEquals(expected_prompt, formatted_prompt) self.assertEqual(expected_prompt, formatted_prompt)
tests/models/vipllava/test_processor_vipllava.py
View file @ 85a1269e
...@@ -38,4 +38,4 @@ class LlavaProcessorTest(unittest.TestCase): ...@@ -38,4 +38,4 @@ class LlavaProcessorTest(unittest.TestCase):
] ]
formatted_prompt = processor.apply_chat_template(messages, add_generation_prompt=True) formatted_prompt = processor.apply_chat_template(messages, add_generation_prompt=True)
self.assertEquals(expected_prompt, formatted_prompt) self.assertEqual(expected_prompt, formatted_prompt)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment