Browse Source

Merge remote-tracking branch 'origin/feat/fill_mask' into feat/nlp

master
智丞 3 years ago
parent
commit
be17f2a74b
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      tests/pipelines/test_fill_mask.py

+ 2
- 2
tests/pipelines/test_fill_mask.py View File

@@ -35,7 +35,7 @@ class FillMaskTest(unittest.TestCase):
'[MASK]. Your [MASK] universe is just a mirror [MASK] of your story.'
}

@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_by_direct_model_download(self):
# sbert
for language in ['zh', 'en']:
@@ -115,7 +115,7 @@ class FillMaskTest(unittest.TestCase):
f'\nori_text: {self.ori_texts[language]}\ninput: {self.test_inputs[language]}\npipeline: '
f'{pipeline_ins(self.test_inputs[language])}\n')

@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_default_model(self):
pipeline_ins = pipeline(task=Tasks.fill_mask)
language = 'en'


Loading…
Cancel
Save