From aa2b4d90bb4d4fe940ab37e11776aca2ea8b5969 Mon Sep 17 00:00:00 2001 From: Mateusz Klimaszewski <mk.klimaszewski@gmail.com> Date: Mon, 9 Nov 2020 12:59:44 +0100 Subject: [PATCH] Fix console prediction. --- combo/main.py | 3 +++ combo/predict.py | 3 +++ 2 files changed, 6 insertions(+) diff --git a/combo/main.py b/combo/main.py index c7aac87..44ad091 100644 --- a/combo/main.py +++ b/combo/main.py @@ -146,6 +146,9 @@ def run(_): else: use_dataset_reader = FLAGS.conllu_format predictor = _get_predictor() + if FLAGS.input_file == "-": + use_dataset_reader = False + predictor.without_sentence_embedding = True if use_dataset_reader: predictor.line_to_conllu = True if FLAGS.silent: diff --git a/combo/predict.py b/combo/predict.py index ebbb372..b6c7172 100644 --- a/combo/predict.py +++ b/combo/predict.py @@ -32,6 +32,7 @@ class SemanticMultitaskPredictor(predictor.Predictor): self._dataset_reader.generate_labels = False self._dataset_reader.lazy = True self._tokenizer = tokenizer + self.without_sentence_embedding = False self.line_to_conllu = line_to_conllu def __call__(self, sentence: Union[str, List[str], List[List[str]], List[data.Sentence]]): @@ -127,6 +128,8 @@ class SemanticMultitaskPredictor(predictor.Predictor): def dump_line(self, outputs: data.Sentence) -> str: # Check whether serialized (str) tree or token's list # Serialized tree has already separators between lines + if self.without_sentence_embedding: + outputs.sentence_embedding = [] if self.line_to_conllu: return sentence2conllu(outputs, keep_semrel=self._dataset_reader.use_sem).serialize() else: -- GitLab