diff --git a/nerfstudio/data/pixel_samplers.py b/nerfstudio/data/pixel_samplers.py index 1d5a9697e..826656f89 100644 --- a/nerfstudio/data/pixel_samplers.py +++ b/nerfstudio/data/pixel_samplers.py @@ -21,6 +21,7 @@ from typing import Dict, Optional, Union import torch from torchtyping import TensorType + from nerfstudio.utils import profiler diff --git a/nerfstudio/models/nerfacto.py b/nerfstudio/models/nerfacto.py index e21f7c8de..a35e9900c 100644 --- a/nerfstudio/models/nerfacto.py +++ b/nerfstudio/models/nerfacto.py @@ -59,8 +59,7 @@ from nerfstudio.model_components.renderers import ( from nerfstudio.model_components.scene_colliders import NearFarCollider from nerfstudio.model_components.shaders import NormalsShader from nerfstudio.models.base_model import Model, ModelConfig -from nerfstudio.utils import colormaps -from nerfstudio.utils import profiler +from nerfstudio.utils import colormaps, profiler @dataclass