def METHOD_NAME(self):
    print(('Testing %s.%s' % (type(self).__name__, self._testMethodName)))
    self.tmp_dir = tempfile.TemporaryDirectory().name
    if not os.path.exists(self.tmp_dir):
        os.makedirs(self.tmp_dir)
    self.model_id = 'damo/cv_ddcolor_image-colorization'
    self.cache_path = snapshot_download(self.model_id)
    self.config = Config.from_file(
        os.path.join(self.cache_path, ModelFile.CONFIGURATION))
    dataset_train = MsDataset.load(
        'imagenet-val5k-image',
        namespace='damo',
        subset_name='default',
        split='validation',
        download_mode=DownloadMode.REUSE_DATASET_IF_EXISTS)._hf_ds
    dataset_val = MsDataset.load(
        'imagenet-val5k-image',
        namespace='damo',
        subset_name='default',
        split='validation',
        download_mode=DownloadMode.REUSE_DATASET_IF_EXISTS)._hf_ds
    self.dataset_train = ImageColorizationDataset(
        dataset_train, self.config.dataset, is_train=True)
    self.dataset_val = ImageColorizationDataset(
        dataset_val, self.config.dataset, is_train=False)