diff --git a/nerfstudio/scripts/render.py b/nerfstudio/scripts/render.py index d9d815ed71..79cc6c7bd9 100644 --- a/nerfstudio/scripts/render.py +++ b/nerfstudio/scripts/render.py @@ -61,6 +61,10 @@ VanillaDataManager, VanillaDataManagerConfig, ) +from nerfstudio.data.datamanagers.parallel_datamanager import ParallelDataManager +from nerfstudio.data.datamanagers.random_cameras_datamanager import ( + RandomCamerasDataManager, +) from nerfstudio.data.datasets.base_dataset import Dataset from nerfstudio.data.scene_box import OrientedBox from nerfstudio.data.utils.dataloaders import FixedIndicesEvalDataloader @@ -666,7 +670,14 @@ def main(self) -> None: install_checks.check_ffmpeg_installed() - assert isinstance(pipeline.datamanager, VanillaDataManager) + assert isinstance( + pipeline.datamanager, + ( + VanillaDataManager, + ParallelDataManager, + RandomCamerasDataManager, + ), + ) steps = int(self.frame_rate * self.seconds) camera_start = pipeline.datamanager.eval_dataloader.get_camera(image_idx=0).flatten() camera_path = get_spiral_path(camera_start, steps=steps, radius=self.radius)