From 33fa0a6be132f838b09177bc2846213d2324d5db Mon Sep 17 00:00:00 2001
From: Franck Dary <franck.dary@lis-lab.fr>
Date: Mon, 28 Jun 2021 10:46:02 +0200
Subject: [PATCH] Corrected bug in decode
---
main.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/main.py b/main.py
index 4912d85..e1bb9de 100755
--- a/main.py
+++ b/main.py
@@ -113,7 +113,7 @@ if __name__ == "__main__" :
transNames = json.load(open(args.model+"/transitions.json", "r"))[1]
args.predictedStr = transInfos[0]
args.predicted = set({colName for colName in args.predictedStr.split(',')})
- transitionSets = [[Transition(elem) for elem in transNames]for ts in transNames]
+ transitionSets = [[Transition(elem) for elem in ts] for ts in transNames]
strategy = json.load(open(args.model+"/strategy.json", "r"))
printTS(transitionSets, sys.stderr)
Decode.decodeMode(args.debug, args.corpus, args.type, transitionSets, strategy, args.reward, args.predicted, args.model)
--
GitLab