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

Summary:
Addresses a sub-task of pytorch#44752.

Pull Request resolved: pytorch#44832

Reviewed By: malfet

Differential Revision: D23744260

Pulled By: samestep

fbshipit-source-id: 46aede57b4fa66a770d5df382b0aea2bd6772b9b

Author: samestep

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
 

torch/testing/_internal/dist_utils.py

@@ -6,7 +6,7 @@
 
 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():