Fix `tests/repo_utils/test_get_test_info.py` (#23485)
fix Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
This commit is contained in:
parent
9728f1134b
commit
3658488ff7
|
@ -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):
|
||||||
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):
|
||||||
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"],
|
||||||
|
|
Loading…
Reference in New Issue