Add Any return annotation to __getattr__ methods that return a union of types. (#150204)

rchen152 · pytorchmergebot · commit c65de03196ae · 2025-04-02T05:25:07.000Z
Adds an `Any` return type annotation to `__getattr__` methods in `torch/_ops.py` that return a union of types. Attribute access returning a union of types can cause issues downstream because consumers would need to handle all of the possible types to make the type checker happy. This doesn't seem to matter today for mypy, presumably because `Any` is always inferred when a return type annotation is missing, but it still makes explicit what mypy is already doing implicitly. Pull Request resolved: #150204 Approved by: https://github.com/malfet
diff --git a/torch/_ops.py b/torch/_ops.py
@@ -1086,7 +1086,7 @@ def _schemas(self):
             for overload_name in self._overload_names
         }
 
-    def __getattr__(self, key):
+    def __getattr__(self, key) -> Any:
         # It is not a valid op_name when __file__ is passed in
         if key == "__file__":
             return "torch.ops"
@@ -1246,7 +1246,7 @@ def __init__(self, name):
     def __iter__(self):
         return iter(self._dir)
 
-    def __getattr__(self, op_name):
+    def __getattr__(self, op_name) -> Any:
         # It is not a valid op_name when __file__ is passed in
         if op_name == "__file__":
             return "torch.ops"
