Fix PreemptedAtPriority in fairshare preemption (#4025)

Signed-off-by: JamesMurkin <[email protected]>
armadaproject · Nov 5, 2024 · 0f87b66 · 0f87b66
1 parent 0c82392
commit 0f87b66
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/internal/scheduler/nodedb/nodedb.go b/internal/scheduler/nodedb/nodedb.go
@@ -753,9 +753,9 @@ func (nodeDb *NodeDb) selectNodeForJobWithFairPreemption(txn *memdb.Txn, jctx *c
 				return nil, errors.WithStack(err)
 			}
 
-			priority, ok := nodeDb.GetScheduledAtPriority(evictedJctx.JobId)
+			priority, ok := nodeDb.GetScheduledAtPriority(job.JobSchedulingContext.JobId)
 			if !ok {
-				priority = evictedJctx.Job.PriorityClass().Priority
+				priority = job.JobSchedulingContext.Job.PriorityClass().Priority
 			}
 			if priority > maxPriority {
 				maxPriority = priority