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

Enable type check for torch.testing._internal.dist_utils.* (#44832)

Browse Source 
Summary:
Addresses a sub-task of https://github.com/pytorch/pytorch/issues/44752.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/44832

Reviewed By: malfet

Differential Revision: D23744260

Pulled By: samestep

fbshipit-source-id: 46aede57b4fa66a770d5df382b0aea2bd6772b9b
This commit is contained in:
Sam Estep 2020-09-18 12:48:38 -07:00 committed by Facebook GitHub Bot
parent 7bd8a6913d
commit c2cf6efd96
2 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mypy.ini
View File

@ -56,9 +56,6 @@ ignore_errors = True
[mypy-torch.testing._internal.te_utils.*]
ignore_errors = True
[mypy-torch.testing._internal.dist_utils.*]
ignore_errors = True
[mypy-torch.testing._internal.codegen.*]
ignore_errors = True

2
torch/testing/_internal/dist_utils.py
View File

@ -6,7 +6,7 @@ import sys
import torch.distributed as dist
import torch.distributed.rpc as rpc
from torch.distributed.rpc import _rref_context_get_debug_info
from torch.distributed.rpc import _rref_context_get_debug_info # type: ignore[attr-defined]
if not dist.is_available():