summaryrefslogtreecommitdiff
path: root/lib
diff options
context:
space:
mode:
Diffstat (limited to 'lib')
-rwxr-xr-xlib/automata.py8
1 files changed, 4 insertions, 4 deletions
diff --git a/lib/automata.py b/lib/automata.py
index a947a61..3386761 100755
--- a/lib/automata.py
+++ b/lib/automata.py
@@ -30,8 +30,8 @@ class Transition:
arguments = [], param_update_function = None,
arg_to_param_map = None, set_param = None):
self.name = name
- self.origin = orig_state.name
- self.destination = dest_state.name
+ self.origin = orig_state
+ self.destination = dest_state
self.energy = energy
self.energy_function = energy_function
self.duration = duration
@@ -74,8 +74,8 @@ class Transition:
def to_json(self):
ret = {
'name' : self.name,
- 'origin' : self.origin,
- 'destination' : self.destination,
+ 'origin' : self.origin.name,
+ 'destination' : self.destination.name,
'is_interrupt' : self.is_interrupt,
'arguments' : self.arguments,
'arg_to_param_map' : self.arg_to_param_map,