try inspect.get_annotations

python · Jul 21, 2024 · 896b68a · 896b68a
1 parent e22b5b3
commit 896b68a
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/mypy/stubgenc.py b/mypy/stubgenc.py
@@ -848,10 +848,7 @@ def generate_class_stub(self, class_name: str, cls: type, output: list[str]) ->
                 attrs.append((attr, value))
 
         # Gets annotations if they exist
-        try:
-            annotations = cls.__annotations__
-        except AttributeError:
-            annotations = {}
+        annotations = inspect.get_annotations(cls)
 
         for attr, value in attrs:
             if attr == "__hash__" and value is None:
@@ -910,10 +907,7 @@ def generate_variable_stub(self, name: str, obj: object, output: list[str]) -> N
         self.record_name(name)
 
         # Gets annotations if they exist
-        try:
-            annotations = self.module.__annotations__
-        except AttributeError:
-            annotations = {}
+        annotations = inspect.get_annotations(self.module)
 
         if name in annotations:
             type_str = self.strip_or_import(annotations[name])