diff --git a/spacy/pipeline/transition_parser.pyx b/spacy/pipeline/transition_parser.pyx index 79e089065..c5591a9f3 100644 --- a/spacy/pipeline/transition_parser.pyx +++ b/spacy/pipeline/transition_parser.pyx @@ -240,6 +240,7 @@ class Parser(TrainablePipe): def predict(self, docs): if isinstance(docs, Doc): docs = [docs] + self._ensure_labels_are_added(docs) if not any(len(doc) for doc in docs): result = self.moves.init_batch(docs) return result @@ -248,14 +249,14 @@ class Parser(TrainablePipe): return states_or_beams def greedy_parse(self, docs, drop=0.): - # Deprecated + # TODO: Deprecated self._resize() with _change_attrs(self.model, beam_width=1): states, _ = self.model.predict((docs, self.moves)) return states def beam_parse(self, docs, int beam_width, float drop=0., beam_density=0.): - # Deprecated + # TODO: Deprecated self._resize() with _change_attrs(self.model, beam_width=self.cfg["beam_width"], beam_density=self.cfg["beam_density"]): beams, _ = self.model.predict((docs, self.moves)) diff --git a/spacy/tests/parser/test_add_label.py b/spacy/tests/parser/test_add_label.py index 540b00f89..4c775a913 100644 --- a/spacy/tests/parser/test_add_label.py +++ b/spacy/tests/parser/test_add_label.py @@ -135,6 +135,7 @@ def test_ner_labels_added_implicitly_on_beam_parse(): assert "D" in ner.labels +@pytest.mark.skip(reason="greedy_parse is deprecated") def test_ner_labels_added_implicitly_on_greedy_parse(): nlp = Language() ner = nlp.add_pipe("beam_ner")