Use introspection plots in Graph.visualize

Does still use networkx based plots if given an input dictionary (and networkx is available), falls introspection plot if networkx is unavailable

Author: ksunden

data_prototype/conversion_edge.py

@@ -372,7 +372,20 @@ def edges(self):
 returnSequenceEdge.from_edges("eval", out_edges, output)
 
 defvisualize(self, input: dict[str, Desc] |None=None):
-importnetworkxasnx
+ifinputisNone:
+from .introspectionimportdraw_graph
+
+draw_graph(self)
+return
+
+try:
+importnetworkxasnx
+exceptImportError:
+from .introspectionimportdraw_graph
+
+draw_graph(self)
+return
+
 importmatplotlib.pyplotasplt
 
 frompprintimportpformat
@@ -382,38 +395,26 @@ def node_format(x):
 
 G=nx.DiGraph()
 
-ifinputisnotNone:
-for_, edgesinself._subgraphs:
-q: list[dict[str, Desc]] = [input]
-explored: set[tuple[tuple[str, str], ...]] =set()
-explored.add(
-tuple(sorted(((k, v.coordinates) fork, vinq[0].items())))
- )
-G.add_node(node_format(q[0]))
-whileq:
-n=q.pop()
-foreinedges:
-ifDesc.compatible(n, e.input):
-w=n|e.output
-ifnode_format(w) notinG:
-G.add_node(node_format(w))
-explored.add(
-tuple(
-sorted(
- ((k, v.coordinates) fork, vinw.items())
- )
- )
+for_, edgesinself._subgraphs:
+q: list[dict[str, Desc]] = [input]
+explored: set[tuple[tuple[str, str], ...]] =set()
+explored.add(tuple(sorted(((k, v.coordinates) fork, vinq[0].items()))))
+G.add_node(node_format(q[0]))
+whileq:
+n=q.pop()
+foreinedges:
+ifDesc.compatible(n, e.input):
+w=n|e.output
+ifnode_format(w) notinG:
+G.add_node(node_format(w))
+explored.add(
+tuple(
+sorted(((k, v.coordinates) fork, vinw.items()))
  )
-q.append(w)
-ifnode_format(w) !=node_format(n):
-G.add_edge(node_format(n), node_format(w), name=e.name)
-else:
-# don't bother separating subgraphs,as the end result is exactly the same here
-foredgeinself._edges:
-G.add_edge(
-node_format(edge.input), node_format(edge.output), name=edge.name
- )
-
+ )
+q.append(w)
+ifnode_format(w) !=node_format(n):
+G.add_edge(node_format(n), node_format(w), name=e.name)
 try:
 pos=nx.shell_layout(G)
 exceptException: