diff --git a/tests/repo_utils/test_get_test_info.py b/tests/repo_utils/test_get_test_info.py index e432dd945ee219..a6d4a9984d321e 100644 --- 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"],