From a7d40fa0f40885d10e5579fd475ab57dc54a27b8 Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Wed, 7 Jul 2021 13:46:49 +0200
Subject: [PATCH] Fixed oracle decode

---
 Decode.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/Decode.py b/Decode.py
index 96b205c..d28ed5f 100644
--- a/Decode.py
+++ b/Decode.py
@@ -26,13 +26,14 @@ def randomDecode(ts, strat, config, debug=False) :
 ################################################################################
 
 ################################################################################
-def oracleDecode(ts, strat, config, debug=False) :
+def oracleDecode(transitionSets, strat, config, debug=False) :
   EOS = Transition("EOS")
   config.moveWordIndex(0)
   config.state = 0
   moved = True
   while moved :
     missingLinks = getMissingLinks(config)
+    ts = transitionSets[config.state] 
     candidates = sorted([[trans.getOracleScore(config, missingLinks), trans] for trans in ts if trans.appliable(config)])
     if len(candidates) == 0 :
       break
-- 
GitLab