From 8298a3c31d9319a25c3d17091de2619e0a91afee Mon Sep 17 00:00:00 2001 From: "xingjun.wxj" Date: Wed, 8 Mar 2023 16:23:58 +0800 Subject: [PATCH] add force redownload for nerf dataset in ut add force redownload for nerf dataset in ut Link: https://code.alibaba-inc.com/Ali-MaaS/MaaS-lib/codereview/11915271 --- tests/pipelines/test_nerf_recon_acc.py | 9 ++++++--- tests/trainers/test_nerf_recon_acc_trainer.py | 8 ++++++-- 2 files changed, 12 insertions(+), 5 deletions(-) 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,