Update root property in the case of nest namespace

aiida_workgraph/utils/__init__.py

@@ -261,7 +261,10 @@ def merge_properties(wgdata: Dict[str, Any]) -> None:
             if "." in key and prop["value"] not in [None, {}]:
                 root, key = key.split(".", 1)
                 root_prop = task["inputs"][root]["property"]
-                update_nested_dict(root_prop["value"], key, prop["value"])
+                # update the root property
+                root_prop["value"] = update_nested_dict(
+                    root_prop["value"], key, prop["value"]
+                )
                 prop["value"] = None
 
 

tests/test_tasks.py

@@ -55,3 +55,19 @@ def test_task_wait(decorated_add: Callable) -> None:
     wg.run()
     report = get_workchain_report(wg.process, "REPORT")
     assert "tasks ready to run: add1" in report
+
+
+def test_set_inputs(decorated_add: Callable) -> None:
+    """Test setting inputs of a task."""
+
+    wg = WorkGraph(name="test_set_inputs")
+    add1 = wg.add_task(decorated_add, "add1", x=1)
+    add1.set({"y": 2, "metadata.store_provenance": False})
+    data = wg.prepare_inputs(metadata=None)
+    assert data["wg"]["tasks"]["add1"]["inputs"]["y"]["property"]["value"] == 2
+    assert (
+        data["wg"]["tasks"]["add1"]["inputs"]["metadata"]["property"]["value"][
+            "store_provenance"
+        ]
+        is False
+    )

tests/test_workgraph.py

@@ -132,6 +132,7 @@ def test_pause_task_before_submit(wg_calcjob):
     # assert wg.tasks["add2"].outputs["sum"].value == 9
 
 
+@pytest.mark.skip(reason="pause task is not stable for the moment.")
 def test_pause_task_after_submit(wg_calcjob):
     wg = wg_calcjob
     wg.tasks["add1"].set({"metadata.options.sleep": 5})