From fd7fd38da04892657ec6f30e18cbe6ed772af2bd Mon Sep 17 00:00:00 2001 From: "mulin.lyh" Date: Fri, 10 Feb 2023 10:14:24 +0800 Subject: [PATCH] fix failed case --- tests/pipelines/test_plug_dialogue.py | 2 +- tests/trainers/test_finetune_mgeo.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/pipelines/test_plug_dialogue.py b/tests/pipelines/test_plug_dialogue.py index cde65bea..4d76b61c 100644 --- a/tests/pipelines/test_plug_dialogue.py +++ b/tests/pipelines/test_plug_dialogue.py @@ -27,7 +27,7 @@ class PlugDialogueTest(unittest.TestCase, DemoCompatibilityCheck): self.model_id = 'damo/plug-dialogue' self.model_revision = 'v1.0.1' - @unittest.skipUnless(test_level() >= 0, 'skip test in current test level') + @unittest.skipUnless(test_level() >= 1, 'skip test in current test level') def test_run_with_pipeline(self): pipeline_ins = pipeline( task=self.task, diff --git a/tests/trainers/test_finetune_mgeo.py b/tests/trainers/test_finetune_mgeo.py index 41f0eb14..bb32e412 100644 --- a/tests/trainers/test_finetune_mgeo.py +++ b/tests/trainers/test_finetune_mgeo.py @@ -50,7 +50,7 @@ class TestFinetuneMGeo(unittest.TestCase): results_files = os.listdir(self.tmp_dir) self.assertIn(f'{trainer.timestamp}.log.json', results_files) - @unittest.skipUnless(test_level() >= 0, 'skip test in current test level') + @unittest.skipUnless(test_level() >= 1, 'skip test in current test level') def test_finetune_geotes_rerank(self): def cfg_modify_fn(cfg): @@ -154,7 +154,7 @@ class TestFinetuneMGeo(unittest.TestCase): output_dir = os.path.join(self.tmp_dir, ModelFile.TRAIN_OUTPUT_DIR) print(f'model is saved to {output_dir}') - @unittest.skipUnless(test_level() >= 0, 'skip test in current test level') + @unittest.skipUnless(test_level() >= 1, 'skip test in current test level') def test_finetune_geoeag(self): def cfg_modify_fn(cfg): @@ -214,7 +214,7 @@ class TestFinetuneMGeo(unittest.TestCase): output_dir = os.path.join(self.tmp_dir, ModelFile.TRAIN_OUTPUT_DIR) print(f'model is saved to {output_dir}') - @unittest.skipUnless(test_level() >= 0, 'skip test in current test level') + @unittest.skipUnless(test_level() >= 1, 'skip test in current test level') def test_finetune_geoeta(self): def cfg_modify_fn(cfg):