Apply static solution to counts

src/aiidalab_qe/app/result/components/status/tree.py

@@ -76,6 +76,13 @@ def _render(self):
         self.children = [
             self.collapse_button,
             self.trunk,
+            ipw.HTML(
+                value="*workflow will re-submit failed calculations",
+                layout=ipw.Layout(
+                    margin="5px 0 0",
+                    font_style="italic",
+                ),
+            ),
         ]
 
     def _update(self):
@@ -94,6 +101,7 @@ class ProcessTreeNode(ipw.VBox, t.Generic[ProcessNodeType]):
         "QeAppWorkChain": "Quantum ESPRESSO app workflow",
         "BandsWorkChain": "Electronic band structure workflow",
         "PwBandsWorkChain": "Electronic band structure workflow",
+        "ProjwfcBandsWorkChain": "Electronic band structure workflow",
         "PwRelaxWorkChain": "Structural relaxation workflow",
         "PdosWorkChain": "Projected density of states workflow",
         "DosCalculation": "Compute density of states",
@@ -116,7 +124,7 @@ class ProcessTreeNode(ipw.VBox, t.Generic[ProcessNodeType]):
             "PwCalculation": "Compute bands",
         },
         "relax": {
-            "PwBaseWorkChain": "Relaxation workflow",
+            "PwBaseWorkChain": "Structure relaxation workflow",
             "PwCalculation": "Optimize structure geometry",
         },
     }
@@ -128,8 +136,6 @@ def __init__(
         on_inspect: t.Callable[[str], None] | None = None,
         **kwargs,
     ):
-        # if not (node and isinstance(node, orm.ProcessNode)):
-        #     raise ValueError("Process node required")
         self.uuid = node.uuid
         self.level = level
         self.on_inspect = on_inspect
@@ -304,12 +310,11 @@ def _add_branches(self, node=None):
                 self.pks.add(child.pk)
 
     def _get_tally(self):
-        total = self._get_current_total(self.node)
-        total = max(total, self.expected_jobs["count"])
+        total = self.expected_jobs["count"]
         dynamic = self.expected_jobs["dynamic"]
         finished = self._count_finished(self.node)
         tally = f"{finished}/{total}"
-        tally += "*" if dynamic and total != finished else ""
+        tally += "*" if dynamic else ""
         tally += " job" if total == 1 else " jobs"
         return tally
 
@@ -335,7 +340,6 @@ def _get_expected(self, inputs: dict[str, dict]) -> dict:
         expected = {}
         count = 0
         dynamic = False
-        has_dynamic_workflows = False
 
         for key, sub_inputs in inputs.items():
             if "metadata" in sub_inputs and "options" in sub_inputs["metadata"]:
@@ -345,13 +349,12 @@ def _get_expected(self, inputs: dict[str, dict]) -> dict:
             elif key != "metadata":
                 # This is a workflow
                 nested = self._get_expected(sub_inputs)
-                has_dynamic_workflows = nested["dynamic"]
                 count += nested["count"]
                 expected[key] = nested
 
         expected |= {
             "count": count,
-            "dynamic": dynamic or has_dynamic_workflows,
+            "dynamic": dynamic,
         }
 
         return expected