diff options
Diffstat (limited to 'lib/automata.py')
| -rwxr-xr-x | lib/automata.py | 12 |
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) |