fixed syntax error. outputs now missing in our json representation

Author: GeigerJ2

example_workflows/arithmetic/aiida_example.py

@@ -0,0 +1,123 @@
+#!/usr/bin/env python
+# coding: utf-8
+
+# # Aiida
+
+# ## Define workflow with aiida
+
+# In[1]:
+
+
+from python_workflow_definition.aiida import write_workflow_json
+
+from aiida_workgraph import WorkGraph, task
+from aiida import orm, load_profile
+load_profile()
+
+workflow_json_filename =  "aiida_simple.json"
+
+
+# In[2]:
+
+
+from workflow import (
+    get_sum as _get_sum,
+    get_prod_and_div as _get_prod_and_div,
+    get_square as _get_square,
+)
+
+
+# In[3]:
+
+
+wg = WorkGraph("arithmetic")
+
+
+# In[4]:
+
+
+get_prod_and_div_task = wg.add_task(
+    task(outputs=['prod', 'div'])(_get_prod_and_div),
+    x=orm.Float(1),
+    y=orm.Float(2),
+)
+
+
+# In[5]:
+
+
+get_sum_task = wg.add_task(
+    _get_sum,
+    x=get_prod_and_div_task.outputs.prod,
+    y=get_prod_and_div_task.outputs.div,
+)
+
+
+# In[6]:
+
+
+get_square_task = wg.add_task(
+    _get_square,
+    x=get_sum_task.outputs.result,
+)
+
+
+# In[7]:
+
+
+write_workflow_json(wg=wg, file_name=workflow_json_filename)
+
+
+# In[8]:
+
+
+get_ipython().system('cat {workflow_json_filename}')
+
+
+# ## Load Workflow with jobflow
+
+# In[9]:
+
+
+from python_workflow_definition.jobflow import load_workflow_json
+
+
+# In[10]:
+
+
+from jobflow.managers.local import run_locally
+
+
+# In[11]:
+
+
+flow = load_workflow_json(file_name=workflow_json_filename)
+
+
+# In[12]:
+
+
+result = run_locally(flow)
+result
+
+
+# ## Load Workflow with pyiron_base
+
+# In[13]:
+
+
+from python_workflow_definition.pyiron_base import load_workflow_json
+
+
+# In[14]:
+
+
+delayed_object_lst = load_workflow_json(file_name=workflow_json_filename)
+delayed_object_lst[-1].draw()
+
+
+# In[15]:
+
+
+delayed_object_lst[-1].pull()
+

src/python_workflow_definition/aiida.py

@@ -90,12 +90,18 @@ def write_workflow_json(wg: WorkGraph, file_name: str) -> dict:
     node_name_mapping = {}
     data_node_name_mapping = {}
     i = 0
+    GRAPH_LEVEL_NAMES = ['graph_inputs', 'graph_outputs', 'graph_ctx']
+
     for node in wg.tasks:
-        executor = node.get_executor()
+
+        if node.name in GRAPH_LEVEL_NAMES:
+            continue
+
         node_name_mapping[node.name] = i
 
-        callable_name = executor["callable_name"]
-        callable_name = f"{executor['module_path']}.{callable_name}"
+        executor = node.get_executor()
+        callable_name = executor.callable_name
+
         data[NODES_LABEL].append({"id": i, "type": "function", "value": callable_name})
         i += 1
 
@@ -110,6 +116,7 @@ def write_workflow_json(wg: WorkGraph, file_name: str) -> dict:
         link_data[SOURCE_PORT_LABEL] = link_data.pop("from_socket")
         data[EDGES_LABEL].append(link_data)
 
+
     for node in wg.tasks:
         for input in node.inputs:
             # assume namespace is not used as input
@@ -141,6 +148,7 @@ def write_workflow_json(wg: WorkGraph, file_name: str) -> dict:
                         SOURCE_PORT_LABEL: None,
                     }
                 )
+
     data[VERSION_LABEL] = VERSION_NUMBER
     PythonWorkflowDefinitionWorkflow(
         **set_result_node(workflow_dict=update_node_names(workflow_dict=data))