Use _get_annotations rather than branching on class dict entry

python-attrs · Nov 17, 2023 · b0d87e3 · b0d87e3
1 parent 80e8a6b
commit b0d87e3
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/src/attr/_make.py b/src/attr/_make.py
@@ -920,11 +920,11 @@ def _create_slots_class(self):
                 # Clear out function from class to avoid clashing.
                 del cd[name]
 
-            if "__annotations__" in cd:
-                for name, func in cached_properties.items():
-                    annotation = inspect.signature(func).return_annotation
-                    if annotation is not inspect.Parameter.empty:
-                        cd["__annotations__"][name] = annotation
+            class_annotations = _get_annotations(self._cls)
+            for name, func in cached_properties.items():
+                annotation = inspect.signature(func).return_annotation
+                if annotation is not inspect.Parameter.empty:
+                    class_annotations[name] = annotation
 
             original_getattr = cd.get("__getattr__")
             if original_getattr is not None: