Don't assume all jobs share the same keys in their limits

naegelyd-insitro · naegelyd-insitro · commit b57a543f3f8b · 2024-03-04T14:24:01.000-05:00
diff --git a/redun/scheduler.py b/redun/scheduler.py
@@ -1313,7 +1313,7 @@ def _add_job_pending_limits(self, job: Job, eval_args: Tuple[Tuple, dict]) -> No
 
     def _add_limits(self, limits1, limits2):
         return {
-            limit_name: limits1[limit_name] + limits2[limit_name]
+            limit_name: limits1.get(limit_name, 0) + limits2.get(limit_name, 0)
             for limit_name in set(limits1) | set(limits2)
         }
 

Original file line number	Diff line number	Diff line change
`@@ -1313,7 +1313,7 @@ def _add_job_pending_limits(self, job: Job, eval_args: Tuple[Tuple, dict]) -> No`
`1313`	`1313`
`1314`	`1314`	`def _add_limits(self, limits1, limits2):`
`1315`	`1315`	`return {`
`1316`		`- limit_name: limits1[limit_name] + limits2[limit_name]`
	`1316`	`+ limit_name: limits1.get(limit_name, 0) + limits2.get(limit_name, 0)`
`1317`	`1317`	`for limit_name in set(limits1) \| set(limits2)`
`1318`	`1318`	`}`
`1319`	`1319`