update from_json / to_json

1 files changed, 3 insertions, 2 deletions

diff --git a/lib/automata.py b/lib/automata.py
index 379b4dc..37b52e1 100755
--- a/lib/automata.py
+++ b/lib/automata.py
@@ -292,7 +292,7 @@ class PTA:
             return cls.from_legacy_json(json_input)
 
         kwargs = dict()
-        for key in ('state_names', 'parameters', 'initial_param_values'):
+        for key in ('state_names', 'parameters', 'initial_param_values', 'accepting_states'):
             if key in json_input:
                 kwargs[key] = json_input[key]
         pta = cls(**kwargs)
@@ -375,7 +375,8 @@ class PTA:
             'parameters' : self.parameters,
             'initial_param_values' : self.initial_param_values,
             'state' : dict([[state.name, state.to_json()] for state in self.state.values()]),
-            'transitions' : [trans.to_json() for trans in self.transitions]
+            'transitions' : [trans.to_json() for trans in self.transitions],
+            'accepting_states' : self.accepting_states,
         }
         return ret