Bug fix in returning word scores.

browsermt · May 15, 2020 · c0bd70c · c0bd70c
1 parent 603256a
commit c0bd70c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/service/api/rapidjson_utils.cpp b/src/service/api/rapidjson_utils.cpp
@@ -111,7 +111,7 @@ Value hyp2json(const marian::Result& nbestlist_item,
 
   if (opts.withWordScores) {
     Value ws(kArrayType);
-    std::vector<float> wscores = hyp->getScoreBreakdown();
+    std::vector<float> wscores = hyp->tracebackWordScores();
     if (service.isRight2LeftDecoder())
       std::reverse(wscores.begin(), wscores.end());
     ws.Reserve(wscores.size(),alloc);