explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-11-10 19:57:17 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix span <-> underscore import cycle

Browse Source
This commit is contained in:
Daniël de Kok 2023-06-26 12:43:21 +02:00
parent bf92ca4f10
commit 122f3b32ad
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
spacy/tokens/underscore.py
View File

@ -3,10 +3,10 @@ import functools
from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple, Union
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
@ -40,7 +40,10 @@ 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 type(obj) == Span: # We used to check if obj is a span, however, this introduces an
# import cycle between the span and underscore modeles. So we
# do a structural type check instead.
if hasattr(obj, "id") and hasattr(obj, "label") and hasattr(obj, "kb_id"):
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)