From 8c7c34d4f4c87fb6e4c01547fc0235dc2b0a1ff2 Mon Sep 17 00:00:00 2001 From: kadarakos Date: Thu, 1 Jun 2023 13:09:25 +0000 Subject: [PATCH] use the 'spans_key' variable name everywhere --- spacy/pipeline/span_finder.py | 6 +++--- spacy/tests/pipeline/test_span_finder.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/spacy/pipeline/span_finder.py b/spacy/pipeline/span_finder.py index a7ee61784..a9874f724 100644 --- a/spacy/pipeline/span_finder.py +++ b/spacy/pipeline/span_finder.py @@ -321,7 +321,7 @@ class SpanFinder(TrainablePipe): @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""" def span_finder_suggester( @@ -333,8 +333,8 @@ def build_span_finder_suggester(candidates_key: str) -> Suggester: lengths = [] for doc in docs: length = 0 - if doc.spans[candidates_key]: - for span in doc.spans[candidates_key]: + if doc.spans[spans_key]: + for span in doc.spans[spans_key]: spans.append([span.start, span.end]) length += 1 diff --git a/spacy/tests/pipeline/test_span_finder.py b/spacy/tests/pipeline/test_span_finder.py index f9a8b3a7d..ee2e10f87 100644 --- a/spacy/tests/pipeline/test_span_finder.py +++ b/spacy/tests/pipeline/test_span_finder.py @@ -204,7 +204,7 @@ def test_span_finder_suggester(): span_finder.set_annotations(docs, span_finder.predict(docs)) suggester = registry.misc.get("spacy.span_finder_suggester.v1")( - candidates_key=SPANS_KEY + spans_key=SPANS_KEY ) candidates = suggester(docs)