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

adjust types and improve setting new attributes

Browse Source
This commit is contained in:
thomashacker 2022-09-28 13:00:07 +02:00
parent 064ba9d2bc
commit 534dd59421
1 changed files with 5 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
spacy/tokens/underscore.py
View File

@ -2,10 +2,10 @@ from typing import Dict, Any, List, Optional, Tuple, Union, TYPE_CHECKING
import functools import functools
import copy import copy
from ..errors import Errors from ..errors import Errors
from .span import Span
if TYPE_CHECKING: if TYPE_CHECKING:
from .doc import Doc from .doc import Doc
from .span import Span
from .token import Token from .token import Token
@ -25,9 +25,9 @@ class Underscore:
obj: Union["Doc", "Span", "Token"], obj: Union["Doc", "Span", "Token"],
start: Optional[int] = None, start: Optional[int] = None,
end: Optional[int] = None, end: Optional[int] = None,
label: Optional[Union[str, int]] = None, label: int = 0,
kb_id: Optional[Union[str, int]] = None, kb_id: int = 0,
span_id: Optional[Union[str, int]] = None, span_id: int = 0,
): ):
object.__setattr__(self, "_extensions", extensions) object.__setattr__(self, "_extensions", extensions)
object.__setattr__(self, "_obj", obj) object.__setattr__(self, "_obj", obj)
@ -39,10 +39,7 @@ class Underscore:
object.__setattr__(self, "_doc", obj.doc) object.__setattr__(self, "_doc", obj.doc)
object.__setattr__(self, "_start", start) object.__setattr__(self, "_start", start)
object.__setattr__(self, "_end", end) object.__setattr__(self, "_end", end)
if label is not None or kb_id is not None or span_id is not None: if type(obj) == Span:
# It is reasonably safe to assume that if label, kb_id and span_id are None,
# they were not passed to this function. Span.label and kb_id are
# converted to 0 in Span's c-tor and setters if they are None
object.__setattr__(self, "_label", label) object.__setattr__(self, "_label", label)
object.__setattr__(self, "_kb_id", kb_id) object.__setattr__(self, "_kb_id", kb_id)
object.__setattr__(self, "_span_id", span_id) object.__setattr__(self, "_span_id", span_id)