yingda.chen 3 years ago
parent
commit
e7571a566f
3 changed files with 5 additions and 5 deletions
  1. +1
    -1
      tests/pipelines/test_image_matting.py
  2. +1
    -1
      tests/pipelines/test_text_classification.py
  3. +3
    -3
      tests/pydatasets/test_py_dataset.py

+ 1
- 1
tests/pipelines/test_image_matting.py View File

@@ -60,7 +60,7 @@ class ImageMattingTest(unittest.TestCase):
cv2.imwrite('result.png', result['output_png'])
print(f'Output written to {osp.abspath("result.png")}')

@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_modelscope_dataset(self):
dataset = PyDataset.load('beans', split='train', target='image')
img_matting = pipeline(Tasks.image_matting, model=self.model_id)


+ 1
- 1
tests/pipelines/test_text_classification.py View File

@@ -87,7 +87,7 @@ class SequenceClassificationTest(unittest.TestCase):
result = text_classification(dataset)
self.printDataset(result)

@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_modelscope_dataset(self):
text_classification = pipeline(task=Tasks.text_classification)
# loaded from modelscope dataset


+ 3
- 3
tests/pydatasets/test_py_dataset.py View File

@@ -33,7 +33,7 @@ class ImgPreprocessor(Preprocessor):

class PyDatasetTest(unittest.TestCase):

@unittest.skipUnless(test_level() >= 1, 'skip test in current test level')
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
def test_ds_basic(self):
ms_ds_full = PyDataset.load('squad')
ms_ds_full_hf = hfdata.load_dataset('squad')
@@ -49,7 +49,7 @@ class PyDatasetTest(unittest.TestCase):
print(next(iter(ms_ds_train)))
print(next(iter(ms_image_train)))

@unittest.skipUnless(test_level() >= 1, 'skip test in current test level')
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@require_torch
def test_to_torch_dataset_text(self):
model_id = 'damo/bert-base-sst2'
@@ -64,7 +64,7 @@ class PyDatasetTest(unittest.TestCase):
dataloader = torch.utils.data.DataLoader(pt_dataset, batch_size=5)
print(next(iter(dataloader)))

@unittest.skipUnless(test_level() >= 1, 'skip test in current test level')
@unittest.skipUnless(test_level() >= 2, 'skip test in current test level')
@require_tf
def test_to_tf_dataset_text(self):
import tensorflow as tf


Loading…
Cancel
Save