From 28770ef736fba36f2bf4952e1bdf93fa4db500e2 Mon Sep 17 00:00:00 2001 From: Brandon Thomas Date: Mon, 4 Dec 2023 05:00:46 -0500 Subject: [PATCH] fix query issue --- .../generic_inference/web/kill_generic_inference_jobs.rs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/crates/lib/mysql_queries/src/queries/generic_inference/web/kill_generic_inference_jobs.rs b/crates/lib/mysql_queries/src/queries/generic_inference/web/kill_generic_inference_jobs.rs index 4928fcd6f4..5ac8ae9155 100644 --- a/crates/lib/mysql_queries/src/queries/generic_inference/web/kill_generic_inference_jobs.rs +++ b/crates/lib/mysql_queries/src/queries/generic_inference/web/kill_generic_inference_jobs.rs @@ -81,7 +81,7 @@ WHERE status IN( separated.push_unseparated(") "); if let Some(priority) = maybe_priority_or_lower { - query_builder.push(format!(" AND priority <= {priority} ")); + query_builder.push(format!(" AND priority_level <= {priority} ")); } match target { @@ -136,7 +136,7 @@ mod tests { let statuses = HashSet::from([KillableStatus::Failed]); let builder = query_builder(statuses, KillableTarget::AllJobs, Some(10)); let query = cleanup_query(&builder.into_sql()); - assert_eq!(query, "UPDATE generic_inference_jobs SET status='cancelled_by_system' WHERE status IN( ?) AND priority <= 10".to_string()); + assert_eq!(query, "UPDATE generic_inference_jobs SET status='cancelled_by_system' WHERE status IN( ?) AND priority_level <= 10".to_string()); } #[test]