Yingda Chen
3 years ago
1 changed files with 2 additions and 2 deletions
Unified View
Diff Options
+ 2
- 2
tests/pipelines/test_text2text_generation.py
View File
| @@ -30,7 +30,7 @@ class Text2TextGenerationTest(unittest.TestCase, DemoCompatibilityCheck): | |||||
| f'pipeline1: {pipeline1(self.input)}\npipeline2: {pipeline2(self.input)}' | f'pipeline1: {pipeline1(self.input)}\npipeline2: {pipeline2(self.input)}' | ||||
| ) | ) | ||||
| @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_pipeline_with_model_instance(self): | def test_run_pipeline_with_model_instance(self): | ||||
| model = Model.from_pretrained(self.model_id) | model = Model.from_pretrained(self.model_id) | ||||
| preprocessor = Text2TextGenerationPreprocessor(model.model_dir) | preprocessor = Text2TextGenerationPreprocessor(model.model_dir) | ||||
| @@ -40,7 +40,7 @@ class Text2TextGenerationTest(unittest.TestCase, DemoCompatibilityCheck): | |||||
| preprocessor=preprocessor) | preprocessor=preprocessor) | ||||
| print(pipeline_ins(self.input)) | print(pipeline_ins(self.input)) | ||||
| @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_pipeline_with_model_id(self): | def test_run_pipeline_with_model_id(self): | ||||
| pipeline_ins = pipeline( | pipeline_ins = pipeline( | ||||
| task=Tasks.text2text_generation, model=self.model_id) | task=Tasks.text2text_generation, model=self.model_id) | ||||