Also handle "CANCELLED+" state

NL-BioImaging · Apr 3, 2024 · 756ea35 · 756ea35
1 parent 359a820
commit 756ea35
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/biomero/slurm_client.py b/biomero/slurm_client.py
@@ -60,7 +60,14 @@ def wait_for_completion(self, slurmClient, omeroConn) -> str:
         Returns:
             str: The final state of the Slurm job.
         """
-        while self.job_state not in ("FAILED", "COMPLETED", "CANCELLED", "TIMEOUT"):
+        while self.job_state not in ("FAILED", 
+                                     "COMPLETED", 
+                                     "CANCELLED",
+                                     "TIMEOUT",
+                                     "FAILED+", 
+                                     "COMPLETED+", 
+                                     "CANCELLED+",
+                                     "TIMEOUT+"):
             job_status_dict, poll_result = slurmClient.check_job_status(
                 [self.job_id])
             if not poll_result.ok: