teaching/muMLE
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Added correct printing of type nodes

Browse source
This commit is contained in:
Andrei Bondarenko 2021-07-13 05:13:55 +02:00
parent 128b3ec1e4
commit 1ba6aecdb9
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
state/devstate.py
View file

@ -28,8 +28,11 @@ class DevState(PyState):
f.write("digraph main {\n") f.write("digraph main {\n")
for n in sorted(self.nodes): for n in sorted(self.nodes):
if n in self.values: 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" % ( f.write("\"a_%s\" [label=\"%s\"];\n" % (
n, str(self.values[n]).replace('"', '\\"'))) n, str(x).replace('"', '\\"')))
else: else:
f.write("\"a_%s\" [label=\"\"];\n" % n) f.write("\"a_%s\" [label=\"\"];\n" % n)
for i, e in sorted(list(self.edges.items())): for i, e in sorted(list(self.edges.items())):