explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-08-05 21:00:19 +03:00
Code Issues Packages Projects Releases Wiki Activity

More type corrections

Browse Source
This commit is contained in:
Richard Hudson 2021-12-23 17:35:47 +01:00
parent 106fb53509
commit 2cae470180
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
spacy/visualization.py
View File

@ -1,5 +1,5 @@
from os import linesep from os import linesep
from typing import List, Optional, Union, cast from typing import Dict, List, Optional, Union, cast
import wasabi import wasabi
from spacy.tokens import Span, Token, Doc from spacy.tokens import Span, Token, Doc
@ -54,8 +54,8 @@ class AttributeFormat:
fg_color: Union[str, int, None] = None, fg_color: Union[str, int, None] = None,
bg_color: Union[str, int, None] = None, bg_color: Union[str, int, None] = None,
permitted_values: Optional[tuple] = None, permitted_values: Optional[tuple] = None,
value_dependent_fg_colors: Optional[dict[str, Union[str, int]]] = None, value_dependent_fg_colors: Optional[Dict[str, Union[str, int]]] = None,
value_dependent_bg_colors: Optional[dict[str, Union[str, int]]] = None, value_dependent_bg_colors: Optional[Dict[str, Union[str, int]]] = None,
): ):
""" """
attribute: the token attribute, e.g. lemma_, ._.holmes.lemma attribute: the token attribute, e.g. lemma_, ._.holmes.lemma
@ -363,7 +363,7 @@ class Visualizer:
] ]
def render_table( def render_table(
self, doc: Doc, columns: list[AttributeFormat], spacing: int self, doc: Doc, columns: List[AttributeFormat], spacing: int
) -> str: ) -> str:
"""Renders a document as a table. """Renders a document as a table.
TODO: specify a specific portion of the document to display. TODO: specify a specific portion of the document to display.