set result node

Author: jan-janssen

python_workflow_definition/src/python_workflow_definition/aiida.py

@@ -11,6 +11,7 @@
     convert_nodes_list_to_dict,
     update_node_names,
     remove_result,
+    set_result_node,
     NODES_LABEL,
     EDGES_LABEL,
     SOURCE_LABEL,
@@ -137,6 +138,6 @@ def write_workflow_json(wg: WorkGraph, file_name: str) -> dict:
                 )
     with open(file_name, "w") as f:
         # json.dump({"nodes": data[], "edges": edges_new_lst}, f)
-        json.dump(update_node_names(workflow_dict=data), f, indent=2)
+        json.dump(set_result_node(workflow_dict=update_node_names(workflow_dict=data)), f, indent=2)
 
     return data

python_workflow_definition/src/python_workflow_definition/jobflow.py

@@ -13,6 +13,7 @@
     update_node_names,
     convert_nodes_list_to_dict,
     remove_result,
+    set_result_node,
     NODES_LABEL,
     EDGES_LABEL,
     SOURCE_LABEL,
@@ -333,9 +334,9 @@ def write_workflow_json(flow: Flow, file_name: str = "workflow.json"):
 
     with open(file_name, "w") as f:
         json.dump(
-            update_node_names(
+            set_result_node(workflow_dict=update_node_names(
                 workflow_dict={NODES_LABEL: nodes_store_lst, EDGES_LABEL: edges_lst}
-            ),
+            )),
             f,
             indent=2,
         )

python_workflow_definition/src/python_workflow_definition/pyiron_base.py

@@ -13,6 +13,7 @@
     convert_nodes_list_to_dict,
     update_node_names,
     remove_result,
+    set_result_node,
     NODES_LABEL,
     EDGES_LABEL,
     SOURCE_LABEL,
@@ -294,9 +295,9 @@ def write_workflow_json(
 
     with open(file_name, "w") as f:
         json.dump(
-            update_node_names(
+            set_result_node(workflow_dict=update_node_names(
                 workflow_dict={NODES_LABEL: nodes_store_lst, EDGES_LABEL: edges_new_lst}
-            ),
+            )),
             f,
             indent=2,
         )