fix plot

Author: jan-janssen

python_workflow_definition/src/python_workflow_definition/plot.py

@@ -39,7 +39,10 @@ def plot(file_name: str):
                     k + "=result[" + v[SOURCE_PORT_LABEL] + "]"
                 )
         for k, v in edge_label_dict.items():
-            graph.add_edge(str(k), str(target_node), label=", ".join(v))
+            if len(v) == 1 and v[0] is not None:
+                graph.add_edge(str(k), str(target_node), label=", ".join(v))
+            else:
+                graph.add_edge(str(k), str(target_node))
 
     svg = nx.nx_agraph.to_agraph(graph).draw(prog="dot", format="svg")
     display(SVG(svg))

python_workflow_definition/src/python_workflow_definition/shared.py

@@ -42,7 +42,8 @@ def get_source_handles(edges_lst: list) -> dict:
 
 def convert_nodes_list_to_dict(nodes_list: list) -> dict:
     return {
-        str(el["id"]): el["value"] for el in sorted(nodes_list, key=lambda d: d["id"])
+        str(el["id"]): el["value"] if "value" in el else el["name"]
+        for el in sorted(nodes_list, key=lambda d: d["id"])
     }