Update results section controls layout on process change

aiidalab · Nov 11, 2024 · 578a997 · 578a997
1 parent 1043a8c
commit 578a997
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/aiidalab_qe/app/result/__init__.py b/src/aiidalab_qe/app/result/__init__.py
@@ -234,3 +234,6 @@ def _update_state(self):
         elif process_node.is_finished_ok:
             self.state = self.State.SUCCESS
             self._model.process_info = PROCESS_COMPLETED
+        if self.state in (self.State.SUCCESS, self.State.FAIL):
+            self._update_kill_button_layout()
+            self._update_clean_scratch_button_layout()
diff --git a/tests/test_result.py b/tests/test_result.py
@@ -3,7 +3,7 @@
 from bs4 import BeautifulSoup
 
 from aiidalab_qe.app.main import App
-from aiidalab_qe.app.result.summary import WorkChainSummary, WorkChainSummaryModel
+from aiidalab_qe.app.result.summary import WorkChainSummaryModel
 from aiidalab_qe.app.result.viewer import WorkChainViewer