Skip to content

GitLab

Unverified Commit 3658488f authored by Yih-Dar's avatar Yih-Dar Committed by GitHub
Browse files

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



fix
Co-authored-by: default avatarydshieh <ydshieh@users.noreply.github.com>
parent 9728f113
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
tests/repo_utils/test_get_test_info.py
View file @ 3658488f
...@@ -45,7 +45,7 @@ class GetTestInfoTester(unittest.TestCase): ...@@ -45,7 +45,7 @@ class GetTestInfoTester(unittest.TestCase):
"BlipTextImageModelTest": "BlipTextImageModelsModelTester", "BlipTextImageModelTest": "BlipTextImageModelsModelTester",
"BlipTextModelTest": "BlipTextModelTester", "BlipTextModelTest": "BlipTextModelTester",
"BlipTextRetrievalModelTest": "BlipTextRetrievalModelTester", "BlipTextRetrievalModelTest": "BlipTextRetrievalModelTester",
"BlipVQAModelTest": "BlipModelTester", "BlipVQAModelTest": "BlipVQAModelTester",
"BlipVisionModelTest": "BlipVisionModelTester", "BlipVisionModelTest": "BlipVisionModelTester",
} }
...@@ -71,7 +71,7 @@ class GetTestInfoTester(unittest.TestCase): ...@@ -71,7 +71,7 @@ class GetTestInfoTester(unittest.TestCase):
EXPECTED_BLIP_MAPPING = { EXPECTED_BLIP_MAPPING = {
"BlipForConditionalGeneration": ["BlipTextImageModelTest"], "BlipForConditionalGeneration": ["BlipTextImageModelTest"],
"BlipForImageTextRetrieval": ["BlipTextRetrievalModelTest"], "BlipForImageTextRetrieval": ["BlipTextRetrievalModelTest"],
"BlipForQuestionAnswering": ["BlipTextImageModelTest", "BlipVQAModelTest"], "BlipForQuestionAnswering": ["BlipVQAModelTest"],
"BlipModel": ["BlipModelTest"], "BlipModel": ["BlipModelTest"],
"BlipTextModel": ["BlipTextModelTest"], "BlipTextModel": ["BlipTextModelTest"],
"BlipVisionModel": ["BlipVisionModelTest"], "BlipVisionModel": ["BlipVisionModelTest"],
...@@ -99,7 +99,7 @@ class GetTestInfoTester(unittest.TestCase): ...@@ -99,7 +99,7 @@ class GetTestInfoTester(unittest.TestCase):
EXPECTED_BLIP_MAPPING = { EXPECTED_BLIP_MAPPING = {
"BlipForConditionalGeneration": ["BlipTextImageModelsModelTester"], "BlipForConditionalGeneration": ["BlipTextImageModelsModelTester"],
"BlipForImageTextRetrieval": ["BlipTextRetrievalModelTester"], "BlipForImageTextRetrieval": ["BlipTextRetrievalModelTester"],
"BlipForQuestionAnswering": ["BlipModelTester", "BlipTextImageModelsModelTester"], "BlipForQuestionAnswering": ["BlipVQAModelTester"],
"BlipModel": ["BlipModelTester"], "BlipModel": ["BlipModelTester"],
"BlipTextModel": ["BlipTextModelTester"], "BlipTextModel": ["BlipTextModelTester"],
"BlipVisionModel": ["BlipVisionModelTester"], "BlipVisionModel": ["BlipVisionModelTester"],
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment