Fail if process exits with non-zero status (aiidalab#270)

unkcpz · Sep 20, 2022 · d49023d · d49023d
1 parent 97c4d79
commit d49023d
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/aiidalab_qe/steps.py b/aiidalab_qe/steps.py
@@ -927,9 +927,12 @@ def _update_state(self):
                 ProcessState.WAITING,
             ):
                 self.state = self.State.ACTIVE
-            elif process_state in (ProcessState.EXCEPTED, ProcessState.KILLED):
+            elif (
+                process_state in (ProcessState.EXCEPTED, ProcessState.KILLED)
+                or self.process.is_failed
+            ):
                 self.state = self.State.FAIL
-            elif process_state is ProcessState.FINISHED:
+            elif self.process.is_finished_ok:
                 self.state = self.State.SUCCESS
 
     @traitlets.observe("process")