|
|
|
@@ -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) |
|
|
|
|