Browse Source

Added correct printing of type nodes

Andrei Bondarenko 2 years ago
parent
commit
1ba6aecdb9
1 changed files with 4 additions and 1 deletions
  1. 4 1
      state/devstate.py

+ 4 - 1
state/devstate.py

@@ -28,8 +28,11 @@ class DevState(PyState):
             f.write("digraph main {\n")
             for n in sorted(self.nodes):
                 if n in self.values:
+                    x = self.values[n]
+                    if isinstance(x, dict):
+                        x = f"{x.get('type')}"
                     f.write("\"a_%s\" [label=\"%s\"];\n" % (
-                        n, str(self.values[n]).replace('"', '\\"')))
+                        n, str(x).replace('"', '\\"')))
                 else:
                     f.write("\"a_%s\" [label=\"\"];\n" % n)
             for i, e in sorted(list(self.edges.items())):