Guard against missing nodes

edan-bainglass · Dec 30, 2024 · a5f3156 · a5f3156
1 parent cc50b39
commit a5f3156
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/aiidalab_qe/common/bands_pdos/utils.py b/src/aiidalab_qe/common/bands_pdos/utils.py
@@ -53,6 +53,8 @@ def extract_pdos_output(node: WorkChainNode) -> AttributeDict | None:
     `AttributeDict | None`
         The PDOS output node, if available.
     """
+    if not node:
+        return
     if node.process_label == "QeAppWorkChain" and "pdos" in node.outputs:
         return node.outputs.pdos
     if "dos" in node.outputs and "projwfc" in node.outputs:
@@ -73,6 +75,8 @@ def extract_bands_output(node: WorkChainNode) -> AttributeDict | None:
     `AttributeDict | None`
         The bands output node, if available.
     """
+    if not node:
+        return
     if node.process_label == "QeAppWorkChain" and "bands" in node.outputs:
         outputs = node.outputs.bands
     else: