Fixed state query in jobcontroller.

Matgenix · Aug 21, 2023 · fe03b05 · fe03b05
1 parent 0bf90b5
commit fe03b05
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/jobflow_remote/jobs/jobcontroller.py b/src/jobflow_remote/jobs/jobcontroller.py
@@ -139,7 +139,7 @@ def _build_query_wf(
                 not_in_states = list(Firework.STATE_RANKS.keys())
                 not_in_states.remove("WAITING")
                 query["fws.state"] = {"$nin": not_in_states}
-            if state == FlowState.PAUSED:
+            elif state == FlowState.PAUSED:
                 not_in_states = list(Firework.STATE_RANKS.keys())
                 not_in_states.remove("PAUSED")
                 query["fws.state"] = {"$nin": not_in_states}
@@ -162,6 +162,8 @@ def _build_query_wf(
                         }
                     },
                 ]
+            else:
+                raise RuntimeError("Unknown flow state.")
 
         # at variance with Firework doc, the dates in the Workflow are Date objects
         if start_date: