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

Sai-Suraj-27 · web-flow · commit 85a1269e19af · 2024-07-24T18:00:21.000+01:00
Replaced deprecated unittest method with the correct one.
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)

Original file line number	Diff line number	Diff line change
`@@ -44,4 +44,4 @@ def test_chat_template(self):`
`44`	`44`	`]`
`45`	`45`
`46`	`46`	`formatted_prompt = processor.apply_chat_template(messages, add_generation_prompt=True)`
`47`		`- self.assertEquals(expected_prompt, formatted_prompt)`
	`47`	`+ self.assertEqual(expected_prompt, formatted_prompt)`
Original file line number	Diff line number	Diff line change
`@@ -38,4 +38,4 @@ def test_chat_template(self):`
`38`	`38`	`]`
`39`	`39`
`40`	`40`	`formatted_prompt = processor.apply_chat_template(messages, add_generation_prompt=True)`
`41`		`- self.assertEquals(expected_prompt, formatted_prompt)`
	`41`	`+ self.assertEqual(expected_prompt, formatted_prompt)`