Merge pull request #313 from dapper91/decorators-typing

- computed_attr and computed_element decorators typing fixed.

Author: dapper91

pydantic_xml/fields.py

@@ -1,7 +1,7 @@
 import copy
 import dataclasses as dc
 import typing
-from typing import Any, Callable, Dict, Optional, Union
+from typing import Any, Callable, Dict, Optional, Union, overload
 
 import pydantic as pd
 import pydantic_core as pdc
@@ -266,6 +266,21 @@ def decorator(prop: Any) -> Any:
         return decorator(prop)
 
 
+@overload
+def computed_attr(
+        *,
+        name: Optional[str] = None,
+        ns: Optional[str] = None,
+        **kwargs: Any,
+) -> Callable[[PropertyT], PropertyT]:
+    ...
+
+
+@overload
+def computed_attr(prop: PropertyT) -> PropertyT:
+    ...
+
+
 def computed_attr(
         prop: Optional[PropertyT] = None,
         *,
@@ -285,6 +300,23 @@ def computed_attr(
     return computed_entity(EntityLocation.ATTRIBUTE, prop, path=name, ns=ns, **kwargs)
 
 
+@overload
+def computed_element(
+        *,
+        tag: Optional[str] = None,
+        ns: Optional[str] = None,
+        nsmap: Optional[NsMap] = None,
+        nillable: Optional[bool] = None,
+        **kwargs: Any,
+) -> Callable[[PropertyT], PropertyT]:
+    ...
+
+
+@overload
+def computed_element(prop: PropertyT) -> PropertyT:
+    ...
+
+
 def computed_element(
         prop: Optional[PropertyT] = None,
         *,