From bd0a020a7fbc32503befa33e138763fed665c7f0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=A1=8C=E5=97=94?= Date: Fri, 30 Sep 2022 17:48:51 +0800 Subject: [PATCH] fix tests --- tests/pipelines/test_ofa_tasks.py | 13 ------------- 1 file changed, 13 deletions(-) diff --git a/tests/pipelines/test_ofa_tasks.py b/tests/pipelines/test_ofa_tasks.py index 4bdb394a..d89e5d48 100644 --- a/tests/pipelines/test_ofa_tasks.py +++ b/tests/pipelines/test_ofa_tasks.py @@ -37,19 +37,6 @@ class OfaTasksTest(unittest.TestCase, DemoCompatibilityCheck): result = img_captioning({'image': image}) print(result[OutputKeys.CAPTION]) - @unittest.skipUnless(test_level() >= 1, 'skip test in current test level') - def test_run_with_image_captioning_zh_with_model(self): - model = Model.from_pretrained( - '/apsarapangu/disk2/yichang.zyc/ckpt/MaaS/ofa_image-caption_coco_base_zh' - ) - img_captioning = pipeline( - task=Tasks.image_captioning, - model=model, - ) - image = 'data/test/images/image_captioning.png' - result = img_captioning({'image': image}) - print(result[OutputKeys.CAPTION]) - @unittest.skipUnless(test_level() >= 0, 'skip test in current test level') def test_run_with_image_captioning_with_name(self): img_captioning = pipeline(