423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-08-02 20:02:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

upgrade pyre version in fbcode/vision - batch 2

Browse Source 
Differential Revision: D38448021

fbshipit-source-id: 966708035115b4870a74f0ca0bbf8ea88b853527
This commit is contained in:
Pyre Bot Jr 2022-08-04 22:18:46 -07:00 committed by Facebook GitHub Bot
parent b28754f8e1
commit a0f786f4cf
3 changed files with 1 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pytorch3d/implicitron/models/generic_model.py
View File

@ -532,6 +532,7 @@ class GenericModel(ImplicitronModelBase): # pyre-ignore: 13
return None return None
loss = sum(losses_weighted) loss = sum(losses_weighted)
assert torch.is_tensor(loss) assert torch.is_tensor(loss)
# pyre-fixme[7]: Expected `Optional[Tensor]` but got `int`.
return loss return loss
def visualize( def visualize(

4
pytorch3d/implicitron/tools/video_writer.py
View File

@ -84,8 +84,6 @@ class VideoWriter:
or a 2-tuple defining the size of the output image. or a 2-tuple defining the size of the output image.
""" """
# pyre-fixme[6]: For 1st param expected `Union[PathLike[str], str]` but got
# `Optional[str]`.
outfile = os.path.join(self.cache_dir, self.regexp % self.frame_num) outfile = os.path.join(self.cache_dir, self.regexp % self.frame_num)
if isinstance(frame, matplotlib.figure.Figure): if isinstance(frame, matplotlib.figure.Figure):
@ -127,8 +125,6 @@ class VideoWriter:
video_path: The path to the generated video. video_path: The path to the generated video.
""" """
# pyre-fixme[6]: For 1st param expected `Union[PathLike[str], str]` but got
# `Optional[str]`.
regexp = os.path.join(self.cache_dir, self.regexp) regexp = os.path.join(self.cache_dir, self.regexp)
if self.output_format == "visdom": # works for ppt too if self.output_format == "visdom": # works for ppt too

2
pytorch3d/io/obj_io.py
View File

@ -215,8 +215,6 @@ def load_obj(
""" """
data_dir = "./" data_dir = "./"
if isinstance(f, (str, bytes, Path)): if isinstance(f, (str, bytes, Path)):
# pyre-fixme[6]: For 1st param expected `PathLike[Variable[AnyStr <: [str,
# bytes]]]` but got `Union[Path, bytes, str]`.
data_dir = os.path.dirname(f) data_dir = os.path.dirname(f)
if path_manager is None: if path_manager is None:
path_manager = PathManager() path_manager = PathManager()