diff --git a/tests/pipelines/test_nerf_recon_acc.py b/tests/pipelines/test_nerf_recon_acc.py index bc5ad1b2..95d879fb 100644 --- a/tests/pipelines/test_nerf_recon_acc.py +++ b/tests/pipelines/test_nerf_recon_acc.py @@ -8,7 +8,7 @@ from modelscope.hub.snapshot_download import snapshot_download from modelscope.msdatasets import MsDataset from modelscope.outputs import OutputKeys from modelscope.pipelines import pipeline -from modelscope.utils.constant import Tasks +from modelscope.utils.constant import DownloadMode, Tasks from modelscope.utils.demo_utils import DemoCompatibilityCheck from modelscope.utils.test_utils import test_level @@ -18,8 +18,11 @@ class NeRFReconAccTest(unittest.TestCase, DemoCompatibilityCheck): def setUp(self) -> None: self.model_id = 'damo/cv_nerf-3d-reconstruction-accelerate_damo' data_dir = MsDataset.load( - 'nerf_recon_dataset', namespace='damo', - split='train').config_kwargs['split_config']['train'] + 'nerf_recon_dataset', + namespace='damo', + split='train', + download_mode=DownloadMode.FORCE_REDOWNLOAD + ).config_kwargs['split_config']['train'] nerf_synthetic_dataset = os.path.join(data_dir, 'nerf_synthetic') blender_scene = 'lego' self.data_dir = os.path.join(nerf_synthetic_dataset, blender_scene) diff --git a/tests/trainers/test_nerf_recon_acc_trainer.py b/tests/trainers/test_nerf_recon_acc_trainer.py index 514aa262..4b6c8091 100644 --- a/tests/trainers/test_nerf_recon_acc_trainer.py +++ b/tests/trainers/test_nerf_recon_acc_trainer.py @@ -4,6 +4,7 @@ import unittest from modelscope.msdatasets import MsDataset from modelscope.trainers.cv import NeRFReconAccTrainer +from modelscope.utils.constant import DownloadMode from modelscope.utils.test_utils import test_level @@ -14,8 +15,11 @@ class TestNeRFReconAccTrainer(unittest.TestCase): model_id = 'damo/cv_nerf-3d-reconstruction-accelerate_damo' data_dir = MsDataset.load( - 'nerf_recon_dataset', namespace='damo', - split='train').config_kwargs['split_config']['train'] + 'nerf_recon_dataset', + namespace='damo', + split='train', + download_mode=DownloadMode.FORCE_REDOWNLOAD + ).config_kwargs['split_config']['train'] trainer = NeRFReconAccTrainer( model=model_id,