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

use the 'spans_key' variable name everywhere

Browse Source
This commit is contained in:
kadarakos 2023-06-01 13:09:25 +00:00
parent fe964e7831
commit 8c7c34d4f4
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
spacy/pipeline/span_finder.py
View File

@ -321,7 +321,7 @@ class SpanFinder(TrainablePipe):
@registry.misc("spacy.span_finder_suggester.v1") @registry.misc("spacy.span_finder_suggester.v1")
def build_span_finder_suggester(candidates_key: str) -> Suggester: def build_span_finder_suggester(spans_key: str) -> Suggester:
"""Suggest every candidate predicted by the SpanFinder""" """Suggest every candidate predicted by the SpanFinder"""
def span_finder_suggester( def span_finder_suggester(
@ -333,8 +333,8 @@ def build_span_finder_suggester(candidates_key: str) -> Suggester:
lengths = [] lengths = []
for doc in docs: for doc in docs:
length = 0 length = 0
if doc.spans[candidates_key]: if doc.spans[spans_key]:
for span in doc.spans[candidates_key]: for span in doc.spans[spans_key]:
spans.append([span.start, span.end]) spans.append([span.start, span.end])
length += 1 length += 1

2
spacy/tests/pipeline/test_span_finder.py
View File

@ -204,7 +204,7 @@ def test_span_finder_suggester():
span_finder.set_annotations(docs, span_finder.predict(docs)) span_finder.set_annotations(docs, span_finder.predict(docs))
suggester = registry.misc.get("spacy.span_finder_suggester.v1")( suggester = registry.misc.get("spacy.span_finder_suggester.v1")(
candidates_key=SPANS_KEY spans_key=SPANS_KEY
) )
candidates = suggester(docs) candidates = suggester(docs)