Fix tests/repo_utils/test_get_test_info.py (#23485)

fix Co-authored-by: ydshieh <[email protected]>
huggingface · May 20, 2023 · 3658488 · 3658488
1 parent 9728f11
commit 3658488
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/repo_utils/test_get_test_info.py b/tests/repo_utils/test_get_test_info.py
@@ -45,7 +45,7 @@ def test_get_test_to_tester_mapping(self):
             "BlipTextImageModelTest": "BlipTextImageModelsModelTester",
             "BlipTextModelTest": "BlipTextModelTester",
             "BlipTextRetrievalModelTest": "BlipTextRetrievalModelTester",
-            "BlipVQAModelTest": "BlipModelTester",
+            "BlipVQAModelTest": "BlipVQAModelTester",
             "BlipVisionModelTest": "BlipVisionModelTester",
         }
 
@@ -71,7 +71,7 @@ def test_get_model_to_test_mapping(self):
         EXPECTED_BLIP_MAPPING = {
             "BlipForConditionalGeneration": ["BlipTextImageModelTest"],
             "BlipForImageTextRetrieval": ["BlipTextRetrievalModelTest"],
-            "BlipForQuestionAnswering": ["BlipTextImageModelTest", "BlipVQAModelTest"],
+            "BlipForQuestionAnswering": ["BlipVQAModelTest"],
             "BlipModel": ["BlipModelTest"],
             "BlipTextModel": ["BlipTextModelTest"],
             "BlipVisionModel": ["BlipVisionModelTest"],
@@ -99,7 +99,7 @@ def test_get_model_to_tester_mapping(self):
         EXPECTED_BLIP_MAPPING = {
             "BlipForConditionalGeneration": ["BlipTextImageModelsModelTester"],
             "BlipForImageTextRetrieval": ["BlipTextRetrievalModelTester"],
-            "BlipForQuestionAnswering": ["BlipModelTester", "BlipTextImageModelsModelTester"],
+            "BlipForQuestionAnswering": ["BlipVQAModelTester"],
             "BlipModel": ["BlipModelTester"],
             "BlipTextModel": ["BlipTextModelTester"],
             "BlipVisionModel": ["BlipVisionModelTester"],