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