Discard

midokura · Apr 24, 2024 · 90b0f12 · 90b0f12
1 parent d91d8a8
commit 90b0f12
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/app.py b/src/app.py
@@ -71,7 +71,7 @@ def process_workflow_job():
         node_ids.add(job.node_id)
 
     elif job.action == "in_progress":
-        node_ids.remove(job.node_id)
+        node_ids.discard(job.node_id)
         job_requested = jobs.get(job.id)
         time_to_start = None
         if not job_requested:
@@ -99,7 +99,7 @@ def process_workflow_job():
         jobs[job.id] = job
 
     elif job.action == "completed":
-        node_ids.remove(job.node_id)
+        node_ids.discard(job.node_id)
         job_requested = jobs.get(job.id)
         if not job_requested:
             app.logger.warning(f"Job {job.id} is {job.action} but not stored!")