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

Replaced deprecated unittest method with the correct one.
huggingface · Jul 24, 2024 · 85a1269 · 85a1269
1 parent edd68f4
commit 85a1269
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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
@@ -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)