From 4ddbee9b0824a93497bd5063bcf8b5c6640cca85 Mon Sep 17 00:00:00 2001 From: maturk Date: Thu, 20 Jul 2023 18:23:46 +0300 Subject: [PATCH] type ignore --- nerfstudio/cameras/camera_optimizers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/nerfstudio/cameras/camera_optimizers.py b/nerfstudio/cameras/camera_optimizers.py index e5520599e..ca3f90742 100644 --- a/nerfstudio/cameras/camera_optimizers.py +++ b/nerfstudio/cameras/camera_optimizers.py @@ -113,7 +113,7 @@ class CameraOptimizer(nn.Module): def apply_to_raybundle(self, raybundle: RayBundle) -> None: """Apply the pose correction to the raybundle""" if self.config.mode != "off": - correction_matrices = self(raybundle.camera_indices.squeeze()) + correction_matrices = self(raybundle.camera_indices.squeeze()) # type: ignore raybundle.origins = raybundle.origins + correction_matrices[:, :3, 3] raybundle.directions = torch.bmm(correction_matrices[:, :3, :3], raybundle.directions[..., None]).squeeze()