upgrade pyre version in fbcode/vision - batch 2

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

View File

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

View File

@ -84,8 +84,6 @@ class VideoWriter:
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)
if isinstance(frame, matplotlib.figure.Figure):
@ -127,8 +125,6 @@ class VideoWriter:
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)
if self.output_format == "visdom": # works for ppt too

View File

@ -215,8 +215,6 @@ def load_obj(
"""
data_dir = "./"
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)
if path_manager is None:
path_manager = PathManager()