From 3658488ff77ff8d45101293e749263acf437f4d5 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Sat, 20 May 2023 06:53:10 +0200 Subject: [PATCH] Fix `tests/repo_utils/test_get_test_info.py` (#23485) fix Co-authored-by: ydshieh --- tests/repo_utils/test_get_test_info.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/repo_utils/test_get_test_info.py b/tests/repo_utils/test_get_test_info.py index e432dd945e..a6d4a9984d 100644 --- a/tests/repo_utils/test_get_test_info.py +++ b/tests/repo_utils/test_get_test_info.py @@ -45,7 +45,7 @@ class GetTestInfoTester(unittest.TestCase): "BlipTextImageModelTest": "BlipTextImageModelsModelTester", "BlipTextModelTest": "BlipTextModelTester", "BlipTextRetrievalModelTest": "BlipTextRetrievalModelTester", - "BlipVQAModelTest": "BlipModelTester", + "BlipVQAModelTest": "BlipVQAModelTester", "BlipVisionModelTest": "BlipVisionModelTester", } @@ -71,7 +71,7 @@ class GetTestInfoTester(unittest.TestCase): EXPECTED_BLIP_MAPPING = { "BlipForConditionalGeneration": ["BlipTextImageModelTest"], "BlipForImageTextRetrieval": ["BlipTextRetrievalModelTest"], - "BlipForQuestionAnswering": ["BlipTextImageModelTest", "BlipVQAModelTest"], + "BlipForQuestionAnswering": ["BlipVQAModelTest"], "BlipModel": ["BlipModelTest"], "BlipTextModel": ["BlipTextModelTest"], "BlipVisionModel": ["BlipVisionModelTest"], @@ -99,7 +99,7 @@ class GetTestInfoTester(unittest.TestCase): EXPECTED_BLIP_MAPPING = { "BlipForConditionalGeneration": ["BlipTextImageModelsModelTester"], "BlipForImageTextRetrieval": ["BlipTextRetrievalModelTester"], - "BlipForQuestionAnswering": ["BlipModelTester", "BlipTextImageModelsModelTester"], + "BlipForQuestionAnswering": ["BlipVQAModelTester"], "BlipModel": ["BlipModelTester"], "BlipTextModel": ["BlipTextModelTester"], "BlipVisionModel": ["BlipVisionModelTester"],