automata: Add set_param support in YAML files

1 files changed, 2 insertions, 0 deletions

diff --git a/lib/automata.py b/lib/automata.py
index a6b37af..8a0357a 100755
--- a/lib/automata.py
+++ b/lib/automata.py
@@ -405,6 +405,8 @@ class PTA:
                         arg_to_param_map[argument['parameter']] = i
             if 'argument_combination' in transition:
                 kwargs['argument_combination'] = transition['argument_combination']
+            if 'set_param' in transition:
+                kwargs['set_param'] = transition['set_param']
             if 'is_interrupt' in transition:
                 kwargs['is_interrupt'] = transition['is_interrupt']
             if not 'src' in transition: