Skip to content

Commit 0d05dda

Browse files
handle port references as strings
1 parent f2e6573 commit 0d05dda

1 file changed

Lines changed: 8 additions & 2 deletions

File tree

src/python/convert_to_python.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -110,8 +110,14 @@ def make_edge_data(data: dict) -> list[dict]:
110110

111111
edge["source_var_name"] = node["var_name"]
112112
edge["target_var_name"] = target_node["var_name"]
113-
edge["source_port"] = f"[{output_index}]"
114-
edge["target_port"] = f"[{input_index}]"
113+
if isinstance(output_index, str):
114+
edge["source_port"] = f"['{output_index}']"
115+
else:
116+
edge["source_port"] = f"[{output_index}]"
117+
if isinstance(input_index, str):
118+
edge["target_port"] = f"['{input_index}']"
119+
else:
120+
edge["target_port"] = f"[{input_index}]"
115121
block_to_input_index[target_block] += 1
116122

117123
return data["edges"]

0 commit comments

Comments
 (0)