From 767a8360c7acfee196287068c9f16a93efaf997e Mon Sep 17 00:00:00 2001 From: suluyan Date: Thu, 23 Jun 2022 23:08:24 +0800 Subject: [PATCH] fix --- modelscope/utils/constant.py | 1 - tests/pipelines/test_fill_mask.py | 4 ++-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/modelscope/utils/constant.py b/modelscope/utils/constant.py index 1663cb63..e824db9a 100644 --- a/modelscope/utils/constant.py +++ b/modelscope/utils/constant.py @@ -44,7 +44,6 @@ class Tasks(object): text_generation = 'text-generation' table_question_answering = 'table-question-answering' feature_extraction = 'feature-extraction' - sentence_similarity = 'sentence-similarity' fill_mask = 'fill-mask' summarization = 'summarization' question_answering = 'question-answering' diff --git a/tests/pipelines/test_fill_mask.py b/tests/pipelines/test_fill_mask.py index 87f71b31..49c5dc8a 100644 --- a/tests/pipelines/test_fill_mask.py +++ b/tests/pipelines/test_fill_mask.py @@ -35,7 +35,7 @@ class FillMaskTest(unittest.TestCase): '[MASK]. Your [MASK] universe is just a mirror [MASK] of your story.' } - @unittest.skipUnless(test_level() >= 1, 'skip test in current test level') + @unittest.skipUnless(test_level() >= 2, 'skip test in current test level') def test_run_by_direct_model_download(self): # sbert for language in ['zh', 'en']: @@ -115,7 +115,7 @@ class FillMaskTest(unittest.TestCase): f'\nori_text: {self.ori_texts[language]}\ninput: {self.test_inputs[language]}\npipeline: ' f'{pipeline_ins(self.test_inputs[language])}\n') - @unittest.skipUnless(test_level() >= 0, 'skip test in current test level') + @unittest.skipUnless(test_level() >= 2, 'skip test in current test level') def test_run_with_default_model(self): pipeline_ins = pipeline(task=Tasks.fill_mask) language = 'en'