Merge branch 'main' into nodes

Author: jan-janssen

python_workflow_definition/src/python_workflow_definition/aiida.py

@@ -1,9 +1,11 @@
 from importlib import import_module
-import traceback
-from aiida_workgraph import WorkGraph, task
-from aiida_pythonjob.data.serializer import general_serializer
 import json
+import traceback
+
 from aiida import orm
+from aiida_pythonjob.data.serializer import general_serializer
+from aiida_workgraph import WorkGraph, task
+from aiida_workgraph.socket import TaskSocketNamespace
 
 from python_workflow_definition.shared import convert_nodes_list_to_dict
 
@@ -66,7 +68,6 @@ def load_workflow_json(file_name):
 
 
 def write_workflow_json(wg, file_name):
-    from aiida_workgraph.socket import TaskSocketNamespace
     data = {"nodes": [], "edges": []}
     node_name_mapping = {}
     data_node_name_mapping = {}