Support arg_to_param_map (dict instead of function)

1 files changed, 4 insertions, 3 deletions

diff --git a/lib/automata.py b/lib/automata.py
index 0948cee..ff32bf7 100755
--- a/lib/automata.py
+++ b/lib/automata.py
@@ -51,9 +51,10 @@ class Transition:
         if self.param_update_function:
             return self.param_update_function(param_dict, args)
         if self.arg_to_param_map:
-            ret = {}
-            #for arg_index in range(self.arg_to_param_map):
-            #    if self.arg_to_param_map[arg_index]
+            ret = param_dict.copy()
+            for k, v in self.arg_to_param_map.items():
+                ret[k] = args[v]
+            return ret
         return param_dict
 
 class State: