add event tracking

This commit is contained in:
班扬
2022-11-01 15:51:00 +08:00
parent 63a08e7be6
commit e45ab2c32d

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,