OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Fix typing for setup_context in autograd (#101464)

Browse Source 
The original only matches a tuple of length 1, but it's intended to match any length.

Also, it now aligns with the docstring at L320
d5cba0618a/torch/autograd/function.py (L320)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/101464
Approved by: https://github.com/soulitzer, https://github.com/kit1980
This commit is contained in:
lkct 2023-05-16 18:41:35 +00:00 committed by PyTorch MergeBot
parent eac5f2a8e4
commit e7681b53e3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
torch/autograd/function.py
View File

@ -339,7 +339,7 @@ class _SingleLevelFunction(_C._FunctionBase, FunctionCtx, _HookMixin, metaclass=
" autograd.Function.")
@staticmethod
def setup_context(ctx: Any, inputs: Tuple[Any], output: Any) -> Any:
def setup_context(ctx: Any, inputs: Tuple[Any, ...], output: Any) -> Any:
r"""There are two ways to define the forward pass of an autograd.Function.
Either: