Fix type definition problem

tomschr · tlaferriere · tomschr · commit 0398baaa2fdb · 2021-04-28T08:30:35.000+02:00
Define new type "Decorator" for function "deprecated" to
avoid this mypy error:

src/semver/_deprecated.py:69: error: Incompatible return value type
  (got "Callable[[VarArg(Any), KwArg(Any)], Callable[..., F]]",
   expected "Union[Callable[..., F], partial[Any]]")

Co-authored-by: Thomas Laferriere &lt;tlaferriere@users.noreply.github.com&gt;
diff --git a/src/semver/_deprecated.py b/src/semver/_deprecated.py
@@ -7,19 +7,19 @@
 import warnings
 from functools import partial, wraps
 from types import FrameType
-from typing import Type, Union, Callable, cast
+from typing import Type, Callable, cast
 
 from . import cli
 from .version import Version
-from ._types import F, String
+from ._types import Decorator, F, String
 
 
 def deprecated(
     func: F = None,
     replace: str = None,
     version: str = None,
     category: Type[Warning] = DeprecationWarning,
-) -> Union[Callable[..., F], partial]:
+) -> Decorator:
     """
     Decorates a function to output a deprecation warning.
 
diff --git a/src/semver/_types.py b/src/semver/_types.py
@@ -1,5 +1,6 @@
 """Typing for semver."""
 
+from functools import partial
 from typing import Union, Optional, Tuple, Dict, Iterable, Callable, TypeVar
 
 VersionPart = Union[int, Optional[str]]
@@ -8,3 +9,4 @@
 VersionIterator = Iterable[VersionPart]
 String = Union[str, bytes]
 F = TypeVar("F", bound=Callable)
+Decorator = Union[Callable[..., F], partial]
