diff --git a/spacy/tokens/doc.pyi b/spacy/tokens/doc.pyi index f0cdaee87..7760b4b5e 100644 --- a/spacy/tokens/doc.pyi +++ b/spacy/tokens/doc.pyi @@ -2,7 +2,9 @@ from typing import Callable, Protocol, Iterable, Iterator, Optional from typing import Union, Tuple, List, Dict, Any, overload from cymem.cymem import Pool from thinc.types import Floats1d, Floats2d, Ints2d + from .span import Span +from .span_group import SpanGroup from .token import Token from ._dict_proxies import SpanGroups from ._retokenize import Retokenizer @@ -128,6 +130,7 @@ class Doc: outside: Optional[List[Span]] = ..., default: str = ... ) -> None: ... + ents_spangroup: SpanGroup @property def noun_chunks(self) -> Iterator[Span]: ... @property