From 9471f0a9a7e23c7dad101e935a10b23682694c24 Mon Sep 17 00:00:00 2001 From: Yingda Chen Date: Mon, 1 Aug 2022 13:35:16 +0800 Subject: [PATCH] [to #42322933] disable ner test temporarily --- tests/pipelines/test_named_entity_recognition.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/pipelines/test_named_entity_recognition.py b/tests/pipelines/test_named_entity_recognition.py index 7f99b328..17708afe 100644 --- a/tests/pipelines/test_named_entity_recognition.py +++ b/tests/pipelines/test_named_entity_recognition.py @@ -32,7 +32,7 @@ class NamedEntityRecognitionTest(unittest.TestCase): print() print(f'pipeline2: {pipeline2(input=self.sentence)}') - @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_model_from_modelhub(self): model = Model.from_pretrained(self.model_id) tokenizer = NERPreprocessor(model.model_dir) @@ -42,7 +42,7 @@ class NamedEntityRecognitionTest(unittest.TestCase): preprocessor=tokenizer) print(pipeline_ins(input=self.sentence)) - @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_with_model_name(self): pipeline_ins = pipeline( task=Tasks.named_entity_recognition, model=self.model_id)