Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-19 01:29:24 +01:00
Code Issues Packages Projects Releases Wiki Activity

[to #42322933] test level check

Browse Source 
Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/9143809
This commit is contained in:
yingda.chen
2022-06-23 16:55:48 +08:00
parent 371530ed1d
commit 1a0d4af55a
10 changed files with 19 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tests/pydatasets/test_py_dataset.py
View File

@@ -33,6 +33,7 @@ class ImgPreprocessor(Preprocessor):
class PyDatasetTest(unittest.TestCase):
@unittest.skipUnless(test_level() >= 1, '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')
@@ -82,7 +83,7 @@ class PyDatasetTest(unittest.TestCase):
drop_remainder=True)
print(next(iter(tf_dataset)))
@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_img(self):
ms_image_train = PyDataset.from_hf_dataset(
@@ -94,7 +95,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_img(self):
import tensorflow as tf