Improve subprocess parsing

MycroftAI · Jul 18, 2020 · fc06b6c · fc06b6c
1 parent ae0a9f9
commit fc06b6c
Showing 1 changed file with 14 additions and 3 deletions.
diff --git a/runner/precise_runner/runner.py b/runner/precise_runner/runner.py
@@ -62,9 +62,20 @@ def stop(self):
     def get_prediction(self, chunk):
         if len(chunk) != self.chunk_size:
             raise ValueError('Invalid chunk size: ' + str(len(chunk)))
-        self.proc.stdin.write(chunk)
-        self.proc.stdin.flush()
-        return float(self.proc.stdout.readline())
+        try:
+            self.proc.stdin.write(chunk)
+            self.proc.stdin.flush()
+            line = self.proc.stdout.readline()
+        except BrokenPipeError:
+            raise SystemExit(0)
+        if not line:
+            raise SystemExit(0)
+        else:
+            try:
+                return float(line)
+            except ValueError:
+                output = line + self.proc.stdout.read()
+                raise RuntimeError('Engine produced the following error: {}'.format(output))
 
 
 class ListenerEngine(Engine):