| @@ -32,7 +32,7 @@ class NamedEntityRecognitionTest(unittest.TestCase): | |||||
| print() | print() | ||||
| print(f'pipeline2: {pipeline2(input=self.sentence)}') | 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): | def test_run_with_model_from_modelhub(self): | ||||
| model = Model.from_pretrained(self.model_id) | model = Model.from_pretrained(self.model_id) | ||||
| tokenizer = NERPreprocessor(model.model_dir) | tokenizer = NERPreprocessor(model.model_dir) | ||||
| @@ -42,7 +42,7 @@ class NamedEntityRecognitionTest(unittest.TestCase): | |||||
| preprocessor=tokenizer) | preprocessor=tokenizer) | ||||
| print(pipeline_ins(input=self.sentence)) | 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): | def test_run_with_model_name(self): | ||||
| pipeline_ins = pipeline( | pipeline_ins = pipeline( | ||||
| task=Tasks.named_entity_recognition, model=self.model_id) | task=Tasks.named_entity_recognition, model=self.model_id) | ||||