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

remove allow-untyped-defs for torch/_C/_distributed_autograd.pyi (#143369)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/143369
Approved by: https://github.com/aorenste
This commit is contained in:
bobrenjc93 2024-12-17 07:16:58 -08:00 committed by PyTorch MergeBot
parent 37a1b9efcc
commit 74e66a21b4
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
torch/_C/_distributed_autograd.pyi
View File

@ -1,4 +1,3 @@
# mypy: allow-untyped-defs
from typing import Any
import torch
@ -22,6 +21,6 @@ def _get_debug_info() -> dict[str, str]: ...
def backward(
context_id: int,
roots: list[torch.Tensor],
retain_graph=False,
retain_graph: bool = False,
) -> None: ...
def get_gradients(context_id: int) -> dict[torch.Tensor, torch.Tensor]: ...