Skip to content

Commit 7c8a90c

Browse files
vfdev-5facebook-github-bot
authored andcommitted
Added missing antialias argument to functional.pyi.in (#72420)
Summary: Description: - Added missing antialias argument to functional.pyi.in - mypy is happy if checking `interpolate` method with antialias argument Related torchvision issue: pytorch/vision#5329 Pull Request resolved: #72420 Reviewed By: mruberry Differential Revision: D34043182 Pulled By: albanD fbshipit-source-id: ca64a8f0d2516c1be5b060c1c24e0b1ed2127b96
1 parent a6657a9 commit 7c8a90c

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

torch/nn/functional.pyi.in

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -305,7 +305,8 @@ def upsample(input: Any, size: Optional[Any] = ..., scale_factor: Optional[Any]
305305

306306

307307
def interpolate(input: Any, size: Optional[Any] = ..., scale_factor: Optional[Any] = ..., mode: str = ...,
308-
align_corners: Optional[Any] = ..., recompute_scale_factor: Optional[Any] = ...): ...
308+
align_corners: Optional[Any] = ..., recompute_scale_factor: Optional[Any] = ...,
309+
antialias: bool = ...): ...
309310

310311

311312
def upsample_nearest(input: Any, size: Optional[Any] = ..., scale_factor: Optional[Any] = ...): ...

0 commit comments

Comments
 (0)