fix task from workgraph

src/aiida_workgraph/engine/utils.py

@@ -13,9 +13,11 @@ def prepare_for_workgraph_task(task: dict, kwargs: dict) -> tuple:
         # because kwargs is updated using update_nested_dict_with_special_keys
         # which means the data is grouped by the task name
         for socket_name, value in data.items():
-            wgdata["tasks"][task_name]["inputs"][socket_name]["property"][
-                "value"
-            ] = value
+            input = wgdata["tasks"][task_name]["inputs"][socket_name]
+            if input["identifier"] == "workgraph.namespace":
+                input["value"] = value
+            else:
+                input["property"]["value"] = value
     # merge the properties
     # organize_nested_inputs(wgdata)
     # serialize_workgraph_inputs(wgdata)

tests/test_task_from_workgraph.py

@@ -37,8 +37,8 @@ def test_build_task_from_workgraph(decorated_add: Callable) -> None:
     wg.add_task(decorated_add, name="add2", y=3)
     wg.add_link(add1_task.outputs["result"], wg_task.inputs["add1.x"])
     wg.add_link(wg_task.outputs["add2.result"], wg.tasks["add2"].inputs["x"])
-    assert len(wg_task.inputs) == 21
-    assert len(wg_task.outputs) == 6
-    wg.submit(wait=True)
-    # wg.run()
-    assert wg.tasks["add2"].outputs["result"].socket_value.value == 12
+    assert len(wg_task.inputs) == 3
+    assert len(wg_task.outputs) == 4
+    # wg.submit(wait=True)
+    wg.run()
+    assert wg.tasks.add2.outputs.result.socket_value.value == 12