explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-01-04 06:16:33 +03:00
Code Issues Packages Projects Releases Wiki Activity

Make Doc.activations type visible to MyPy

Browse Source
This commit is contained in:
Daniël de Kok 2022-06-22 17:05:34 +02:00
parent acf47e8e48
commit 42526a6fa1
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
spacy/tokens/doc.pyi
View File

@ -1,7 +1,7 @@
from typing import Callable, Protocol, Iterable, Iterator, Optional from typing import Callable, Protocol, Iterable, Iterator, Optional
from typing import Union, Tuple, List, Dict, Any, overload from typing import Union, Tuple, List, Dict, Any, overload
from cymem.cymem import Pool from cymem.cymem import Pool
from thinc.types import Floats1d, Floats2d, Ints2d from thinc.types import ArrayXd, Floats1d, Floats2d, Ints2d
from .span import Span from .span import Span
from .token import Token from .token import Token
from ._dict_proxies import SpanGroups from ._dict_proxies import SpanGroups
@ -22,6 +22,7 @@ class Doc:
max_length: int max_length: int
length: int length: int
sentiment: float sentiment: float
activations: Dict[str, Dict[str, ArrayXd]]
cats: Dict[str, float] cats: Dict[str, float]
user_hooks: Dict[str, Callable[..., Any]] user_hooks: Dict[str, Callable[..., Any]]
user_token_hooks: Dict[str, Callable[..., Any]] user_token_hooks: Dict[str, Callable[..., Any]]