diff --git a/nerfstudio/cameras/camera_optimizers.py b/nerfstudio/cameras/camera_optimizers.py index 77b7e2b47..df8296ea1 100644 --- a/nerfstudio/cameras/camera_optimizers.py +++ b/nerfstudio/cameras/camera_optimizers.py @@ -103,8 +103,8 @@ class CameraOptimizer(nn.Module): def forward( self, - indices: Int[Tensor, "num_cameras"], - ) -> Float[Tensor, "num_cameras 3 4"]: + indices: Int[Tensor, "camera_indices"], + ) -> Float[Tensor, "camera_indices 3 4"]: """Indexing into camera adjustments. Args: indices: indices of Cameras to optimize.