Fix jobflow loading pyiron_base

.github/workflows/pyiron.yml

@@ -25,4 +25,5 @@ jobs:
         pip install -e python_workflow_definition
         conda install -c conda-forge jupyter papermill
         export ESPRESSO_PSEUDO=$(pwd)/espresso/pseudo
-        papermill universal_qe_to_pyiron_base.ipynb universal_qe_to_pyiron_base_out.ipynb -k "python3"
+        papermill universal_qe_to_pyiron_base.ipynb universal_qe_to_pyiron_base_out.ipynb -k "python3"
+        papermill pyiron_base_to_jobflow.ipynb pyiron_base_to_jobflow_out.ipynb -k "python3"

python_workflow_definition/src/python_workflow_definition/jobflow.py

@@ -164,7 +164,7 @@ def get_attr_helper(obj, source_handle):
                     obj=memory_dict[vw['source']], source_handle=vw['sourceHandle'])
                 for kw, vw in total_dict[k].items()
             }
-            print(k, kwargs)
+            # print(k, kwargs)
             memory_dict[k] = fn(**kwargs)
     return list(memory_dict.values())
 
@@ -193,7 +193,20 @@ def load_workflow_json(file_name):
     with open(file_name, "r") as f:
         content = json.load(f)
 
-    edges_new_lst = content["edges"]
+    edges_new_lst = []
+    for edge in content["edges"]:
+        if edge['sourceHandle'] is None:
+            edges_new_lst.append(edge)
+        else:
+            edges_new_lst.append(
+                {
+                    'target': edge['target'],
+                    'targetHandle': edge['targetHandle'],
+                    'source': edge['source'],
+                    'sourceHandle': str(edge['sourceHandle']),
+                }
+            )
+
     nodes_new_dict = {}
     for k, v in content["nodes"].items():
         if isinstance(v, str) and "." in v: