upgrade pyre version in fbcode/vision - batch 2

Differential Revision: D42947615

fbshipit-source-id: 47b078fdf68567220e15993ab643f85771b0d340

pytorch3d/datasets/r2n2/r2n2.py

@@ -368,6 +368,8 @@ class R2N2(ShapeNetBase):  # pragma: no cover
         T = RT[3, :3]
         return R, T
 
+    # pyre-fixme[14]: `render` overrides method defined in `ShapeNetBase`
+    #  inconsistently.
     def render(
         self,
         model_ids: Optional[List[str]] = None,

pytorch3d/implicitron/evaluation/evaluator.py

@@ -57,6 +57,7 @@ class ImplicitronEvaluator(EvaluatorBase):
     def __post_init__(self):
         run_auto_creation(self)
 
+    # pyre-fixme[14]: `run` overrides method defined in `EvaluatorBase` inconsistently.
     def run(
         self,
         model: ImplicitronModelBase,

pytorch3d/implicitron/models/generic_model.py

@@ -315,6 +315,8 @@ class GenericModel(ImplicitronModelBase):  # pyre-ignore: 13
 
         self.log_loss_weights()
 
+    # pyre-fixme[14]: `forward` overrides method defined in `ImplicitronModelBase`
+    #  inconsistently.
     def forward(
         self,
         *,  # force keyword-only arguments

pytorch3d/implicitron/models/implicit_function/scene_representation_networks.py

@@ -345,6 +345,8 @@ class SRNImplicitFunction(ImplicitFunctionBase, torch.nn.Module):
     def raymarch_function_tweak_args(cls, type, args: DictConfig) -> None:
         args.pop("latent_dim", None)
 
+    # pyre-fixme[14]: `forward` overrides method defined in `ImplicitFunctionBase`
+    #  inconsistently.
     def forward(
         self,
         *,
@@ -404,6 +406,8 @@ class SRNHyperNetImplicitFunction(ImplicitFunctionBase, torch.nn.Module):
         args.pop("latent_dim", None)
         args.pop("latent_dim_hypernet", None)
 
+    # pyre-fixme[14]: `forward` overrides method defined in `ImplicitFunctionBase`
+    #  inconsistently.
     def forward(
         self,
         *,

pytorch3d/implicitron/models/model_dbir.py

@@ -41,6 +41,8 @@ class ModelDBIR(ImplicitronModelBase):
     bg_color: Tuple[float, float, float] = (0.0, 0.0, 0.0)
     max_points: int = -1
 
+    # pyre-fixme[14]: `forward` overrides method defined in `ImplicitronModelBase`
+    #  inconsistently.
     def forward(
         self,
         *,  # force keyword-only arguments

pytorch3d/implicitron/models/renderer/raymarcher.py

@@ -111,6 +111,8 @@ class AccumulativeRaymarcherBase(RaymarcherBase, torch.nn.Module):
             "minimum": lambda curr, acc: torch.minimum(curr, acc),
         }[self.weight_function_type]
 
+    # pyre-fixme[14]: `forward` overrides method defined in `RaymarcherBase`
+    #  inconsistently.
     def forward(
         self,
         rays_densities: torch.Tensor,

pytorch3d/implicitron/models/renderer/sdf_renderer.py

@@ -66,6 +66,8 @@ class SignedDistanceFunctionRenderer(BaseRenderer, torch.nn.Module):  # pyre-ign
     def requires_object_mask(self) -> bool:
         return True
 
+    # pyre-fixme[14]: `forward` overrides method defined in `BaseRenderer`
+    #  inconsistently.
     def forward(
         self,
         ray_bundle: ImplicitronRayBundle,

pytorch3d/implicitron/tools/video_writer.py

@@ -86,6 +86,8 @@ class VideoWriter:
                 or a 2-tuple defining the size of the output image.
         """
 
+        # pyre-fixme[6]: For 1st argument expected `Union[PathLike[str], str]` but
+        #  got `Optional[str]`.
         outfile = os.path.join(self.cache_dir, self.regexp % self.frame_num)
 
         if isinstance(frame, matplotlib.figure.Figure):
@@ -130,6 +132,8 @@ class VideoWriter:
         if self.frame_num == 0:
             return ""
 
+        # pyre-fixme[6]: For 1st argument expected `Union[PathLike[str], str]` but
+        #  got `Optional[str]`.
         regexp = os.path.join(self.cache_dir, self.regexp)
 
         if shutil.which(self.ffmpeg_bin) is None:

pytorch3d/io/obj_io.py

@@ -215,6 +215,8 @@ def load_obj(
     """
     data_dir = "./"
     if isinstance(f, (str, bytes, Path)):
+        # pyre-fixme[6]: For 1st argument expected `PathLike[Variable[AnyStr <:
+        #  [str, bytes]]]` but got `Union[Path, bytes, str]`.
         data_dir = os.path.dirname(f)
     if path_manager is None:
         path_manager = PathManager()
@@ -296,6 +298,8 @@ class MeshObjFormat(MeshFormatInterpreter):
     def __init__(self) -> None:
         self.known_suffixes = (".obj",)
 
+    # pyre-fixme[14]: `read` overrides method defined in `MeshFormatInterpreter`
+    #  inconsistently.
     def read(
         self,
         path: PathOrStr,
@@ -320,6 +324,8 @@ class MeshObjFormat(MeshFormatInterpreter):
         )
         return mesh
 
+    # pyre-fixme[14]: `save` overrides method defined in `MeshFormatInterpreter`
+    #  inconsistently.
     def save(
         self,
         data: Meshes,

pytorch3d/io/off_io.py

@@ -456,6 +456,8 @@ class MeshOffFormat(MeshFormatInterpreter):
         )
         return mesh
 
+    # pyre-fixme[14]: `save` overrides method defined in `MeshFormatInterpreter`
+    #  inconsistently.
     def save(
         self,
         data: Meshes,

pytorch3d/io/ply_io.py

@@ -1372,6 +1372,8 @@ class MeshPlyFormat(MeshFormatInterpreter):
         )
         return mesh
 
+    # pyre-fixme[14]: `save` overrides method defined in `MeshFormatInterpreter`
+    #  inconsistently.
     def save(
         self,
         data: Meshes,
@@ -1452,6 +1454,8 @@ class PointcloudPlyFormat(PointcloudFormatInterpreter):
         )
         return pointcloud
 
+    # pyre-fixme[14]: `save` overrides method defined in
+    #  `PointcloudFormatInterpreter` inconsistently.
     def save(
         self,
         data: Pointclouds,

pytorch3d/renderer/cameras.py

@@ -695,6 +695,8 @@ class FoVPerspectiveCameras(CamerasBase):
         )
         return transform
 
+    # pyre-fixme[14]: `unproject_points` overrides method defined in `CamerasBase`
+    #  inconsistently.
     def unproject_points(
         self,
         xy_depth: torch.Tensor,
@@ -938,6 +940,8 @@ class FoVOrthographicCameras(CamerasBase):
         )
         return transform
 
+    # pyre-fixme[14]: `unproject_points` overrides method defined in `CamerasBase`
+    #  inconsistently.
     def unproject_points(
         self,
         xy_depth: torch.Tensor,
@@ -1145,6 +1149,8 @@ class PerspectiveCameras(CamerasBase):
         )
         return transform
 
+    # pyre-fixme[14]: `unproject_points` overrides method defined in `CamerasBase`
+    #  inconsistently.
     def unproject_points(
         self,
         xy_depth: torch.Tensor,
@@ -1382,6 +1388,8 @@ class OrthographicCameras(CamerasBase):
         )
         return transform
 
+    # pyre-fixme[14]: `unproject_points` overrides method defined in `CamerasBase`
+    #  inconsistently.
     def unproject_points(
         self,
         xy_depth: torch.Tensor,

pytorch3d/renderer/fisheyecameras.py

@@ -347,6 +347,8 @@ class FishEyeCameras(CamerasBase):
         point3d_est[..., :2] = theta.tan()[..., None] / xr_yrNorm[..., None] * xr_yr
         return point3d_est
 
+    # pyre-fixme[14]: `unproject_points` overrides method defined in `CamerasBase`
+    #  inconsistently.
     def unproject_points(
         self,
         xy_depth: torch.Tensor,

pytorch3d/renderer/mesh/textures.py

@@ -491,6 +491,8 @@ class TexturesAtlas(TexturesBase):
         new_tex._num_faces_per_mesh = new_props["_num_faces_per_mesh"]
         return new_tex
 
+    # pyre-fixme[14]: `sample_textures` overrides method defined in `TexturesBase`
+    #  inconsistently.
     def sample_textures(self, fragments, **kwargs) -> torch.Tensor:
         """
         This is similar to a nearest neighbor sampling and involves a
@@ -927,6 +929,8 @@ class TexturesUV(TexturesBase):
         new_tex._num_faces_per_mesh = new_props["_num_faces_per_mesh"]
         return new_tex
 
+    # pyre-fixme[14]: `sample_textures` overrides method defined in `TexturesBase`
+    #  inconsistently.
     def sample_textures(self, fragments, **kwargs) -> torch.Tensor:
         """
         Interpolate a 2D texture map using uv vertex texture coordinates for each
@@ -1450,6 +1454,8 @@ class TexturesVertex(TexturesBase):
         new_tex._num_verts_per_mesh = new_props["_num_verts_per_mesh"]
         return new_tex
 
+    # pyre-fixme[14]: `sample_textures` overrides method defined in `TexturesBase`
+    #  inconsistently.
     def sample_textures(self, fragments, faces_packed=None) -> torch.Tensor:
         """
         Determine the color for each rasterized face. Interpolate the colors for
@@ -1518,6 +1524,8 @@ class TexturesVertex(TexturesBase):
 
         return self.__class__(sub_features)
 
+    # pyre-fixme[14]: `faces_verts_textures_packed` overrides method defined in
+    #  `TexturesBase` inconsistently.
     def faces_verts_textures_packed(self, faces_packed=None) -> torch.Tensor:
         """
         Samples texture from each vertex and for each face in the mesh.