From c2cf6efd9635555026d7bdbccc17ffa2bf639eda Mon Sep 17 00:00:00 2001 From: Sam Estep Date: Fri, 18 Sep 2020 12:48:38 -0700 Subject: [PATCH] Enable type check for torch.testing._internal.dist_utils.* (#44832) 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 --- mypy.ini | 3 --- torch/testing/_internal/dist_utils.py | 2 +- 2 files changed, 1 insertion(+), 4 deletions(-) diff --git a/mypy.ini b/mypy.ini index 5ebca175802..f36f6c8a801 100644 --- a/mypy.ini +++ b/mypy.ini @@ -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 diff --git a/torch/testing/_internal/dist_utils.py b/torch/testing/_internal/dist_utils.py index c3fd5f55745..b88765211df 100644 --- a/torch/testing/_internal/dist_utils.py +++ b/torch/testing/_internal/dist_utils.py @@ -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():