typing for unproject_points

Summary: Fix the base class annotation for unproject_points.

Reviewed By: patricklabatut

Differential Revision: D33281586

fbshipit-source-id: 1c34e8c4b30b359fcb9307507bc778ad3fecf290

Author: bottler

pytorch3d/renderer/cameras.py

@@ -98,7 +98,7 @@ def get_projection_transform(self):
         """
         raise NotImplementedError()
 
-    def unproject_points(self):
+    def unproject_points(self, xy_depth: torch.Tensor, **kwargs):
         """
         Transform input points from camera coodinates (NDC or screen)
         to the world / camera coordinates.

pytorch3d/renderer/implicit/raysampling.py

@@ -549,7 +549,7 @@ def _xy_to_ray_bundle(
     )
 
     # unproject the points
-    unprojected = cameras.unproject_points(to_unproject, from_ndc=True)  # pyre-ignore
+    unprojected = cameras.unproject_points(to_unproject, from_ndc=True)
 
     # split the two planes back
     rays_plane_1_world = unprojected[:, :n_rays_per_image]