班扬
3 years ago
1 changed files with 2 additions and 2 deletions
Unified View
Diff Options
+ 2
- 2
tests/msdatasets/test_dataset_upload.py
View File
| @@ -8,7 +8,8 @@ import zipfile | |||||
| from modelscope.msdatasets import MsDataset | from modelscope.msdatasets import MsDataset | ||||
| from modelscope.msdatasets.utils.dataset_utils import list_dataset_objects | from modelscope.msdatasets.utils.dataset_utils import list_dataset_objects | ||||
| from modelscope.utils import logger as logging | from modelscope.utils import logger as logging | ||||
| from modelscope.utils.constant import DEFAULT_DATASET_REVISION, ModelFile | |||||
| from modelscope.utils.constant import (DEFAULT_DATASET_REVISION, DownloadMode, | |||||
| ModelFile) | |||||
| from modelscope.utils.test_utils import test_level | from modelscope.utils.test_utils import test_level | ||||
| logger = logging.get_logger(__name__) | logger = logging.get_logger(__name__) | ||||
| @@ -104,7 +105,6 @@ class DatasetUploadTest(unittest.TestCase): | |||||
| @unittest.skipUnless(test_level() >= 1, 'skip test in current test level') | @unittest.skipUnless(test_level() >= 1, 'skip test in current test level') | ||||
| def test_ds_download_dir(self): | def test_ds_download_dir(self): | ||||
| from modelscope.utils.constant import DownloadMode | |||||
| test_ds = MsDataset.load( | test_ds = MsDataset.load( | ||||
| self.dataset_name, | self.dataset_name, | ||||
| namespace=self.namespace, | namespace=self.namespace, | ||||