remove enum check as this has been resolved in previous commit

pytransitions · Mar 30, 2020 · f95febe · f95febe
1 parent b44dc89
commit f95febe
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/transitions/extensions/diagrams_pygraphviz.py b/transitions/extensions/diagrams_pygraphviz.py
@@ -224,9 +224,6 @@ def _add_edges(self, transitions, container):
                 container.add_edge(src_name, dst_name, **edge_attr)
 
     def set_node_style(self, state, style):
-        if isinstance(state, Enum):
-            state = state.name
-
         try:
             node = self.fsm_graph.get_node(state)
             style_attr = self.fsm_graph.style_attributes.get('node', {}).get(style)