refactor: Get annotation with the same property on functions

mkdocstrings · Feb 12, 2022 · ecc7bba · ecc7bba
1 parent e225d26
commit ecc7bba
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/griffe/dataclasses.py b/src/griffe/dataclasses.py
@@ -1095,6 +1095,15 @@ def __init__(
         self.returns: str | Name | Expression | None = returns
         self.decorators: list[Decorator] = decorators or []
 
+    @property
+    def annotation(self) -> str | Name | Expression | None:
+        """Return the return annotation.
+
+        Returns:
+            The function return annotation.
+        """
+        return self.returns
+
     def as_dict(self, **kwargs: Any) -> dict[str, Any]:  # type: ignore[override]
         """Return this function's data as a dictionary.