working benchmark generation

1 files changed, 4 insertions, 8 deletions

diff --git a/lib/automata.py b/lib/automata.py
index df8363f..94b3717 100755
--- a/lib/automata.py
+++ b/lib/automata.py
@@ -259,7 +259,7 @@ class PTA:
     def __init__(self, state_names: list = [],
             accepting_states: list = None,
             parameters: list = [], initial_param_values: list = None,
-            instance: str = None, header: str = None):
+            codegen: dict = {}):
         """
         Return a new PTA object.
 
@@ -275,8 +275,7 @@ class PTA:
         self.state = dict([[state_name, State(state_name)] for state_name in state_names])
         self.accepting_states = accepting_states.copy() if accepting_states else None
         self.parameters = parameters.copy()
-        self.instance = instance
-        self.header = header
+        self.codegen = codegen
         if initial_param_values:
             self.initial_param_values = initial_param_values.copy()
         else:
@@ -385,11 +384,8 @@ class PTA:
         if 'states' in yaml_input:
             kwargs['state_names'] = yaml_input['states']
 
-        if 'instance' in yaml_input:
-            kwargs['instance'] = yaml_input['instance']
-
-        if 'header' in yaml_input:
-            kwargs['header'] = yaml_input['header']
+        if 'codegen' in yaml_input:
+            kwargs['codegen'] = yaml_input['codegen']
 
         pta = cls(**kwargs)