title keyword fix for python 2.7 in diagrams

pytransitions · Jan 29, 2016 · c5992e1 · c5992e1
1 parent ce2228c
commit c5992e1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/transitions/extensions/diagrams.py b/transitions/extensions/diagrams.py
@@ -152,7 +152,9 @@ def __setstate__(self, state):
         self.graph = self.get_graph(title=self.title)
         self.set_node_style(self.graph.get_node(self.current_state.name), 'active')
 
-    def __init__(self, *args, title='State Machine', **kwargs):
+    def __init__(self, *args, **kwargs):
+        # remove title from keywords
+        title = kwargs.pop('title', 'State Machine')
         super(MachineGraphSupport, self).__init__(*args, **kwargs)
 
         # Create graph at beginnning