Visualize nodes inputs for ShellJob tasks (#347)

aiidateam · Nov 19, 2024 · 2559c21 · 2559c21
1 parent f03ff55
commit 2559c21
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/aiida_workgraph/utils/__init__.py b/aiida_workgraph/utils/__init__.py
@@ -734,7 +734,9 @@ def workgraph_to_short_json(
             {"name": name, "identifier": input["identifier"]}
             for name, input in task["inputs"].items()
             if name in task["args"]
+            or (task["identifier"].upper() == "SHELLJOB" and name.startswith("nodes."))
         ]
+
         properties = process_properties(task)
         wgdata_short["nodes"][name] = {
             "label": task["name"],