From 1a2c379e9356f1fdddd9d5f28f040dae9e3b8be5 Mon Sep 17 00:00:00 2001 From: svlandeg Date: Tue, 2 Apr 2024 09:32:52 +0200 Subject: [PATCH] rename to _ensure_ents --- spacy/pipeline/entity_linker.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/spacy/pipeline/entity_linker.py b/spacy/pipeline/entity_linker.py index 03cbe4dc5..7ee0676ee 100644 --- a/spacy/pipeline/entity_linker.py +++ b/spacy/pipeline/entity_linker.py @@ -247,7 +247,7 @@ class EntityLinker(TrainablePipe): if not self.use_gold_ents: return scorer(examples, **kwargs) else: - examples = self._augment_examples(examples) + examples = self._ensure_ents(examples) docs = self.pipe( (eg.predicted for eg in examples), ) @@ -257,7 +257,7 @@ class EntityLinker(TrainablePipe): self.scorer = _score_augmented - def _augment_examples(self, examples: Iterable[Example]) -> Iterable[Example]: + def _ensure_ents(self, examples: Iterable[Example]) -> Iterable[Example]: """If use_gold_ents is true, set the gold entities to (a copy of) eg.predicted.""" if not self.use_gold_ents: return examples @@ -311,7 +311,7 @@ class EntityLinker(TrainablePipe): nO = self.kb.entity_vector_length doc_sample = [] vector_sample = [] - examples = self._augment_examples(islice(get_examples(), 10)) + examples = self._ensure_ents(islice(get_examples(), 10)) for eg in examples: doc = eg.x doc_sample.append(doc) @@ -379,7 +379,7 @@ class EntityLinker(TrainablePipe): losses.setdefault(self.name, 0.0) if not examples: return losses - examples = self._augment_examples(examples) + examples = self._ensure_ents(examples) validate_examples(examples, "EntityLinker.update") # make sure we have something to learn from, if not, short-circuit