SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

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

Browse Source 
Replaced deprecated unittest method with the correct one.
This commit is contained in:
Sai-Suraj-27
2024-07-24 22:30:21 +05:30
committed by GitHub
parent edd68f4ed8
commit 85a1269e19
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/llava/test_processor_llava.py
View File

@@ -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)

2
tests/models/llava_next/test_processor_llava_next.py
View File

@@ -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)

2
tests/models/vipllava/test_processor_vipllava.py
View File

@@ -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)