Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2026-07-13 22:08:46 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix failed case

Browse Source
This commit is contained in:
mulin.lyh
2023-02-10 10:14:24 +08:00
parent 585887b17b
commit fd7fd38da0
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/pipelines/test_plug_dialogue.py
View File

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

6
tests/trainers/test_finetune_mgeo.py
View File

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