Add orm.List ad orm.Dict socket and into type_mapping

pyproject.toml

@@ -121,7 +121,8 @@ workgraph = "aiida_workgraph.cli.cmd_workgraph:workgraph"
 "workgraph.aiida_bool" = "aiida_workgraph.properties.builtins:PropertyAiiDABool"
 "workgraph.aiida_int_vector" = "aiida_workgraph.properties.builtins:PropertyAiiDAIntVector"
 "workgraph.aiida_float_vector" = "aiida_workgraph.properties.builtins:PropertyAiiDAFloatVector"
-"workgraph.aiida_aiida_dict" = "aiida_workgraph.properties.builtins:PropertyAiiDADict"
+"workgraph.aiida_list" = "aiida_workgraph.properties.builtins:PropertyAiiDAList"
+"workgraph.aiida_dict" = "aiida_workgraph.properties.builtins:PropertyAiiDADict"
 "workgraph.aiida_structuredata" = "aiida_workgraph.properties.builtins:PropertyStructureData"
 
 [project.entry-points."aiida_workgraph.socket"]
@@ -138,6 +139,8 @@ workgraph = "aiida_workgraph.cli.cmd_workgraph:workgraph"
 "workgraph.aiida_bool" = "aiida_workgraph.sockets.builtins:SocketAiiDABool"
 "workgraph.aiida_int_vector" = "aiida_workgraph.sockets.builtins:SocketAiiDAIntVector"
 "workgraph.aiida_float_vector" = "aiida_workgraph.sockets.builtins:SocketAiiDAFloatVector"
+"workgraph.aiida_list" = "aiida_workgraph.sockets.builtins:SocketAiiDAList"
+"workgraph.aiida_dict" = "aiida_workgraph.sockets.builtins:SocketAiiDADict"
 "workgraph.aiida_structuredata" = "aiida_workgraph.sockets.builtins:SocketStructureData"
 
 

src/aiida_workgraph/decorator.py

@@ -29,6 +29,8 @@
     orm.Float: "workgraph.aiida_float",
     orm.Str: "workgraph.aiida_string",
     orm.Bool: "workgraph.aiida_bool",
+    orm.List: "workgraph.aiida_list",
+    orm.Dict: "workgraph.aiida_dict",
     orm.StructureData: "workgraph.aiida_structuredata",
 }
 

src/aiida_workgraph/properties/builtins.py

@@ -102,6 +102,18 @@ def validate(self, value: any) -> None:
             )
 
 
+class PropertyAiiDAList(TaskProperty):
+    """A new class for List type."""
+
+    identifier: str = "workgraph.aiida_list"
+    allowed_types = (list, orm.List, str, type(None))
+
+    def set_value(self, value: Union[list, orm.List, str] = None) -> None:
+        if isinstance(value, (list)):
+            value = orm.List(list=value)
+        super().set_value(value)
+
+
 class PropertyAiiDADict(TaskProperty):
     """A new class for Dict type."""
 

src/aiida_workgraph/sockets/builtins.py

@@ -71,6 +71,20 @@ class SocketAiiDABool(TaskSocket):
     property_identifier: str = "workgraph.aiida_bool"
 
 
+class SocketAiiDAList(TaskSocket):
+    """AiiDAList socket."""
+
+    identifier: str = "workgraph.aiida_list"
+    property_identifier: str = "workgraph.aiida_list"
+
+
+class SocketAiiDADict(TaskSocket):
+    """AiiDADict socket."""
+
+    identifier: str = "workgraph.aiida_dict"
+    property_identifier: str = "workgraph.aiida_dict"
+
+
 class SocketAiiDAIntVector(TaskSocket):
     """Socket with a AiiDAIntVector property."""
 

tests/test_config.py

@@ -0,0 +1,6 @@
+def test_load_config():
+    from aiida_workgraph.config import load_config
+
+    config = load_config()
+    assert isinstance(config, dict)
+    assert config == {}

tests/test_socket.py

@@ -19,6 +19,8 @@
         (orm.Str, "abc", "workgraph.aiida_string"),
         (orm.Bool, True, "workgraph.aiida_bool"),
         (orm.Bool, "{{variable}}", "workgraph.aiida_bool"),
+        (orm.List, [1, 2, 3], "workgraph.aiida_list"),
+        (orm.Dict, {"a": 1}, "workgraph.aiida_dict"),
     ),
 )
 def test_type_mapping(data_type, data, identifier) -> None:

tests/widget/test_widget.py

@@ -29,4 +29,4 @@ def test_workgraph_task(wg_calcfunction):
     data = wg.tasks["sumdiff2"].to_html()
     assert isinstance(data, IFrame)
     # check _repr_mimebundle_ is working
-    data = wg._repr_mimebundle_()
+    data = wg.tasks["sumdiff2"]._repr_mimebundle_()