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

Make get_clusters_from_doc return spans in order

Browse Source 
There's no guarantee about the order in which SpanGroup keys will come
out, so access them in sorted order when doing comparisons.
This commit is contained in:
Paul O'Leary McCann 2022-07-12 14:07:40 +09:00
parent 64a0bf4460
commit 1baa334b8a
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -147,7 +147,9 @@ def get_clusters_from_doc(doc) -> List[List[Tuple[int, int]]]:
ints are char spans, to be tokenization independent. ints are char spans, to be tokenization independent.
""" """
out = [] out = []
for key, val in doc.spans.items(): keys = sorted(list(doc.spans.keys()))
for key in keys:
val = doc.spans[key]
cluster = [] cluster = []
for span in val: for span in val: