From d745b2b5163e821585d0701ebaf16a38f4d57eab Mon Sep 17 00:00:00 2001 From: bobrenjc93 Date: Sun, 15 Dec 2024 15:48:39 -0800 Subject: [PATCH] remove allow-untyped-defs for distributed/rpc/_testing/__init__.py (#143271) Pull Request resolved: https://github.com/pytorch/pytorch/pull/143271 Approved by: https://github.com/aorenste --- torch/distributed/rpc/_testing/__init__.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/torch/distributed/rpc/_testing/__init__.py b/torch/distributed/rpc/_testing/__init__.py index 8ac1c02f4cee4..0abd737becafb 100644 --- a/torch/distributed/rpc/_testing/__init__.py +++ b/torch/distributed/rpc/_testing/__init__.py @@ -1,9 +1,7 @@ -# mypy: allow-untyped-defs - import torch -def is_available(): +def is_available() -> bool: return hasattr(torch._C, "_faulty_agent_init")