additional pre- and post-processing in nested workflow

Author: GeigerJ2

example_workflows/nested/main.pwd.json

@@ -1,14 +1,21 @@
 {
   "version": "0.1.1",
   "nodes": [
-    { "id": 0, "type": "workflow", "value": "prod_div.json" },
-    { "id": 1, "value": 1, "type": "input", "name": "a" },
-    { "id": 2, "value": 2, "type": "input", "name": "b" },
-    { "id": 3, "type": "output", "name": "final_result" }
+    { "id": 0, "value": 3, "type": "input", "name": "a" },
+    { "id": 1, "value": 2, "type": "input", "name": "b" },
+    { "id": 2, "value": 4, "type": "input", "name": "c" },
+    { "id": 3, "type": "function", "value": "workflow.get_prod_and_div" },
+    { "id": 4, "type": "workflow", "value": "prod_div.json" },
+    { "id": 5, "type": "function", "value": "workflow.get_sum" },
+    { "id": 6, "type": "output", "name": "final_result" }
   ],
   "edges": [
-    { "target": 0, "targetPort": "x", "source": 1, "sourcePort": null },
-    { "target": 0, "targetPort": "y", "source": 2, "sourcePort": null },
-    { "target": 3, "targetPort": "null", "source": 0, "sourcePort": "result" }
+    { "target": 3, "targetPort": "x", "source": 0, "sourcePort": null },
+    { "target": 3, "targetPort": "y", "source": 2, "sourcePort": null },
+    { "target": 4, "targetPort": "x", "source": 3, "sourcePort": "prod" },
+    { "target": 4, "targetPort": "y", "source": 3, "sourcePort": "div" },
+    { "target": 5, "targetPort": "x", "source": 4, "sourcePort": "result" },
+    { "target": 5, "targetPort": "y", "source": 1, "sourcePort": null },
+    { "target": 6, "targetPort": null, "source": 5, "sourcePort": null }
   ]
-}
+}

src/python_workflow_definition/aiida.py

@@ -203,7 +203,6 @@ def load_workflow_json(file_name: str) -> WorkGraph:
     return wg
 
 
-
 def write_workflow_json(wg: WorkGraph, file_name: str) -> dict:
     data = {NODES_LABEL: [], EDGES_LABEL: []}
     node_name_mapping = {}