mirror of
https://github.com/explosion/spaCy.git
synced 2024-09-22 11:59:14 +03:00
30cf9d6a05
* Improve typing hints for Matcher.__call__ * Add typing hints for DependencyMatcher * Add typing hints to underscore extensions * Update Doc.tensor type (requires numpy 1.21) * Fix typing hints for Language.component decorator * Use generic np.ndarray type in Doc to avoid numpy version update * Fix mypy errors * Fix cyclic import caused by Underscore typing hints * Use Literal type from spacy.compat * Update matcher.pyi import format Co-authored-by: Sofie Van Landeghem <svlandeg@users.noreply.github.com> Co-authored-by: Sofie Van Landeghem <svlandeg@users.noreply.github.com>
38 lines
1.0 KiB
Python
38 lines
1.0 KiB
Python
from typing import List, Tuple, Union, Optional, Callable, Any, Dict, overload
|
|
from ..compat import Literal
|
|
from .matcher import Matcher
|
|
from ..vocab import Vocab
|
|
from ..tokens import Doc, Span
|
|
|
|
class PhraseMatcher:
|
|
def __init__(
|
|
self, vocab: Vocab, attr: Optional[Union[int, str]], validate: bool = ...
|
|
) -> None: ...
|
|
def __reduce__(self) -> Any: ...
|
|
def __len__(self) -> int: ...
|
|
def __contains__(self, key: str) -> bool: ...
|
|
def add(
|
|
self,
|
|
key: str,
|
|
docs: List[List[Dict[str, Any]]],
|
|
*,
|
|
on_match: Optional[
|
|
Callable[[Matcher, Doc, int, List[Tuple[Any, ...]]], Any]
|
|
] = ...,
|
|
) -> None: ...
|
|
def remove(self, key: str) -> None: ...
|
|
@overload
|
|
def __call__(
|
|
self,
|
|
doclike: Union[Doc, Span],
|
|
*,
|
|
as_spans: Literal[False] = ...,
|
|
) -> List[Tuple[int, int, int]]: ...
|
|
@overload
|
|
def __call__(
|
|
self,
|
|
doclike: Union[Doc, Span],
|
|
*,
|
|
as_spans: Literal[True],
|
|
) -> List[Span]: ...
|