Fix PreemptedAtPriority in fairshare preemption

Signed-off-by: JamesMurkin <[email protected]>
armadaproject · Oct 25, 2024 · 526ff91 · 526ff91
1 parent 95fa3b2
commit 526ff91
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