From fc449707dfc1809073452006d0bea389df44d628 Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Mon, 29 Jul 2019 15:07:35 +0200
Subject: [PATCH] Corrected backtrack

---
 decoder/src/Decoder.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/decoder/src/Decoder.cpp b/decoder/src/Decoder.cpp
index d1bb706..f27bd8f 100644
--- a/decoder/src/Decoder.cpp
+++ b/decoder/src/Decoder.cpp
@@ -215,7 +215,7 @@ void Decoder::decodeNoBeam()
 
   while (!config.isFinal())
   {
-    config.setCurrentStateName(tm.getCurrentState());
+    config.setCurrentStateName(tm.getCurrentClassifier()->name);
     Dict::currentClassifierName = tm.getCurrentClassifier()->name;
 
     auto weightedActions = tm.getCurrentClassifier()->weightActions(config);
@@ -377,7 +377,7 @@ void Decoder::decodeBeam()
     bool mustContinue = false;
     for (auto & node : beam)
     {
-      node->config.setCurrentStateName(node->tm.getCurrentState());
+      node->config.setCurrentStateName(node->tm.getCurrentClassifier()->name);
       Dict::currentClassifierName = node->tm.getCurrentClassifier()->name;
 
       node->weightedActions = node->tm.getCurrentClassifier()->weightActions(node->config);
-- 
GitLab