From fae982bc8b249b3ded8869facb423dbd00f77024 Mon Sep 17 00:00:00 2001 From: Petr Andriushchenko Date: Wed, 19 Apr 2023 11:43:48 +0200 Subject: [PATCH] uncommit part of the multimodal search test --- misinformation/test/test_multimodal_search.py | 46 +++++++++---------- 1 file changed, 23 insertions(+), 23 deletions(-) diff --git a/misinformation/test/test_multimodal_search.py b/misinformation/test/test_multimodal_search.py index fce3d60..1bfb954 100644 --- a/misinformation/test/test_multimodal_search.py +++ b/misinformation/test/test_multimodal_search.py @@ -200,29 +200,29 @@ dict_image_gradcam_with_itm_for_blip = { "pre_sorted", ), [ - # ( - # device("cpu"), - # "blip2", - # pre_proc_pic_blip2_blip_albef, - # pre_proc_text_blip2_blip_albef, - # pre_extracted_feature_img_blip2, - # pre_extracted_feature_text_blip2, - # simularity_blip2, - # sorted_blip2, - # ), - # pytest.param( - # device("cuda"), - # "blip2", - # pre_proc_pic_blip2_blip_albef, - # pre_proc_text_blip2_blip_albef, - # pre_extracted_feature_img_blip2, - # pre_extracted_feature_text_blip2, - # simularity_blip2, - # sorted_blip2, - # marks=pytest.mark.skipif( - # gpu_is_not_available, reason="gpu_is_not_availible" - # ), - # ), + ( + device("cpu"), + "blip2", + pre_proc_pic_blip2_blip_albef, + pre_proc_text_blip2_blip_albef, + pre_extracted_feature_img_blip2, + pre_extracted_feature_text_blip2, + simularity_blip2, + sorted_blip2, + ), + pytest.param( + device("cuda"), + "blip2", + pre_proc_pic_blip2_blip_albef, + pre_proc_text_blip2_blip_albef, + pre_extracted_feature_img_blip2, + pre_extracted_feature_text_blip2, + simularity_blip2, + sorted_blip2, + marks=pytest.mark.skipif( + gpu_is_not_available, reason="gpu_is_not_availible" + ), + ), ( device("cpu"), "blip",