"...static/git@developer.sourcefind.cn:dadigang/Ventoy.git" did not exist on "598af7f45c8112321b10da3c38e70468e62ea6a6"
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
...@@ -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)
...@@ -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)
...@@ -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