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

Minor fixes

Browse Source
This commit is contained in:
Paul O'Leary McCann 2021-05-17 14:52:30 +09:00
parent 7c42a8c90a
commit 91b111467b
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
spacy/ml/models/coref_util.py
View File

@ -1,6 +1,7 @@
from thinc.types import Ints2d
from spacy.tokens import Doc
from typing import List, Tuple
from typing import List, Tuple, Callable
from ...util import registry
# type alias to make writing this less tedious
MentionClusters = List[List[Tuple[int, int]]]

4
spacy/pipeline/coref.py
View File

@ -73,7 +73,7 @@ def make_coref(
) -> "CoreferenceResolver":
"""Create a CoreferenceResolver component."""
return CoreferenceResolver(nlp.vocab, model, name, span_cluster_prefix)
return CoreferenceResolver(nlp.vocab, model, name, span_cluster_prefix=span_cluster_prefix)
class CoreferenceResolver(TrainablePipe):
@ -88,7 +88,7 @@ class CoreferenceResolver(TrainablePipe):
model: Model,
name: str = "coref",
*,
span_mentions: str,
span_mentions: str = "coref_mentions",
span_cluster_prefix: str,
) -> None:
"""Initialize a coreference resolution component.