diff --git a/spacy/tokens/doc.pyi b/spacy/tokens/doc.pyi index f0b68862c..62a20279c 100644 --- a/spacy/tokens/doc.pyi +++ b/spacy/tokens/doc.pyi @@ -28,8 +28,12 @@ from .underscore import Underscore DOCBIN_ALL_ATTRS: Tuple[str, ...] -class DocMethod(Protocol): - def __call__(self: Doc, *args: Any, **kwargs: Any) -> Any: ... # type: ignore[misc] +DocMethod = Union[ + Callable[[Doc], Any], + Callable[[Doc, Any], Any], + Callable[[Doc, Any, Any], Any], + Callable[[Doc, Any, Any, Any], Any], +] class Doc: vocab: Vocab diff --git a/spacy/tokens/span.pyi b/spacy/tokens/span.pyi index b982eb810..1714f732d 100644 --- a/spacy/tokens/span.pyi +++ b/spacy/tokens/span.pyi @@ -8,8 +8,12 @@ from .doc import Doc from .token import Token from .underscore import Underscore -class SpanMethod(Protocol): - def __call__(self: Span, *args: Any, **kwargs: Any) -> Any: ... # type: ignore[misc] +SpanMethod = Union[ + Callable[[Span], Any], + Callable[[Span, Any], Any], + Callable[[Span, Any, Any], Any], + Callable[[Span, Any, Any, Any], Any], +] class Span: @classmethod diff --git a/spacy/tokens/token.pyi b/spacy/tokens/token.pyi index 435ace527..4bc41d517 100644 --- a/spacy/tokens/token.pyi +++ b/spacy/tokens/token.pyi @@ -9,8 +9,12 @@ from .morphanalysis import MorphAnalysis from .span import Span from .underscore import Underscore -class TokenMethod(Protocol): - def __call__(self: Token, *args: Any, **kwargs: Any) -> Any: ... # type: ignore[misc] +TokenMethod = Union[ + Callable[[Token], Any], + Callable[[Token, Any], Any], + Callable[[Token, Any, Any], Any], + Callable[[Token, Any, Any, Any], Any], +] class Token: i: int