Return scores and hypotheses from the Python APIv0.6.0

1 files changed, 8 insertions, 4 deletions

diff --git a/python/Python.cc b/python/Python.cc
index 0c1ab368..cc4d7646 100644
--- a/python/Python.cc
+++ b/python/Python.cc
@@ -88,10 +88,14 @@ public:
 
     py::list py_results;
     for (const auto& result : results) {
-      py::list temp;
-      for (const auto& token : result.output())
-        temp.append(token);
-      py_results.append(temp);
+      py::list batch;
+      for (size_t i = 0; i < result.num_hypotheses(); ++i) {
+        py::list hyp;
+        for (const auto& token : result.hypotheses()[i])
+          hyp.append(token);
+        batch.append(py::make_tuple(result.scores()[i], hyp));
+      }
+      py_results.append(batch);
     }
 
     return py_results;