From 2d545d4abcef2441c31c6b14f9504c022f6e08f1 Mon Sep 17 00:00:00 2001 From: Petr Andriushchenko Date: Tue, 7 Mar 2023 14:30:48 +0100 Subject: [PATCH] fixed error in test multimodal search, changed ci --- .github/workflows/ci.yml | 9 +-------- misinformation/test/test_multimodal_search.py | 15 ++++++++++++++- 2 files changed, 15 insertions(+), 9 deletions(-) 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,