Merge pull request #88 from wuruoyu/main

Fix a bug in dot backend
google-ai-edge · Jun 21, 2024 · 9e9d170 · 9e9d170
2 parents b999965 + 1cde522
commit 9e9d170
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/example_adapters/graphviz_dot_adapter/src/graphviz_dot_adapter/main.py b/src/example_adapters/graphviz_dot_adapter/src/graphviz_dot_adapter/main.py
@@ -126,10 +126,10 @@ def _gen_graph_edges(self, cur_gv_graph, node_to_graph_node: Dict):
       target_graph_node.incomingEdges.append(
           graph_builder.IncomingEdge(sourceNodeId=source_node))
 
-      # Recursive on subgraphs.
-      subgraphs = cur_gv_graph.get_subgraphs()
-      for subgraph in subgraphs:
-        self._gen_graph_edges(subgraph, node_to_graph_node)
+    # Recursive on subgraphs.
+    subgraphs = cur_gv_graph.get_subgraphs()
+    for subgraph in subgraphs:
+      self._gen_graph_edges(subgraph, node_to_graph_node)
 
   def _create_graph_node(self, node_name: str,
                          node_label: str,