|
@@ -50,7 +50,7 @@ class ModelverseState(object):
|
|
|
f.write("a_%s [label=\"a_%s (%s)\"];\n" % (n, n, self.values[n]))
|
|
|
else:
|
|
|
f.write("a_%s [label=\"a_%s\"];\n" % (n, n))
|
|
|
- for i, e in self.edges.iteritems():
|
|
|
+ for i, e in self.edges.items():
|
|
|
f.write("%s -> %s [label=\"%s\"];\n" % (e[0], e[1], i))
|
|
|
f.write("}")
|
|
|
return self.root
|
|
@@ -87,6 +87,11 @@ class ModelverseState(object):
|
|
|
|
|
|
with gzip.open(filename, 'rb') as f:
|
|
|
for line in f:
|
|
|
+
|
|
|
+ #handle str to bytes
|
|
|
+ if sys.version_info[0] > 2:
|
|
|
+ line = line.decode()
|
|
|
+
|
|
|
element_type, constructor = line.split(None, 1)
|
|
|
name, values = constructor.split("(", 1)
|
|
|
values, _ = values.rsplit(")", 1)
|