diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index c4ae57c..72608d1 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -32,14 +32,7 @@ jobs: - name: Run pytest run: | cd misinformation - python -m pytest -vv test/test_cropposts.py --cov=. --cov-report=xml - python -m pytest -vv test/test_display.py --cov=. --cov-report=xml - python -m pytest -vv test/test_faces.py --cov=. --cov-report=xml - python -m pytest -vv test/test_multimodal_search.py --cov=. --cov-report=xml - python -m pytest -vv test/test_objects.py --cov=. --cov-report=xml - python -m pytest -vv test/test_summary.py --cov=. --cov-report=xml - python -m pytest -vv test/test_text.py -m "not gcv" --cov=. --cov-report=xml - python -m pytest -vv test/test_utils.py --cov=. --cov-report=xml + python -m pytest -vv -m "not gcv" --cov=. --cov-report=xml - name: Upload coverage if: matrix.os == 'ubuntu-22.04' && matrix.python-version == '3.9' uses: codecov/codecov-action@v3 diff --git a/misinformation/test/test_multimodal_search.py b/misinformation/test/test_multimodal_search.py index b13ce29..ace3a42 100644 --- a/misinformation/test/test_multimodal_search.py +++ b/misinformation/test/test_multimodal_search.py @@ -129,6 +129,19 @@ pre_extracted_feature_img_clip = [ 0.020485712215304375, ] +pre_extracted_feature_img_parsing_clip = [ + 0.01621132344007492, + -0.004035486374050379, + -0.04304071143269539, + -0.03459808602929115, + 0.016922621056437492, + -0.025056276470422745, + -0.04178355261683464, + 0.02165347896516323, + -0.003224249929189682, + 0.020485712215304375, +] + pre_extracted_feature_img_clip_vitl14 = [ -0.023943455889821053, -0.021703708916902542, @@ -256,7 +269,7 @@ pre_extracted_feature_text_clip_vitl14_336 = [ ), ), ( - device("cuda"), + device("cpu"), "blip", pre_proc_pic_blip2_blip_albef, pre_proc_text_blip2_blip_albef,