From 29d70ca0ac452653a3a5f499c5d55d8af0a20025 Mon Sep 17 00:00:00 2001
From: Colin Thomas <33940547+colinthomas-z80@users.noreply.github.com>
Date: Thu, 22 Aug 2024 13:26:11 -0400
Subject: [PATCH] set resources in factories (#3924)

---
 batch_job/src/vine_factory.c       | 4 ++++
 batch_job/src/work_queue_factory.c | 4 ++++
 2 files changed, 8 insertions(+)

diff --git a/batch_job/src/vine_factory.c b/batch_job/src/vine_factory.c
index 37b4bbc3c0..b397a7d277 100644
--- a/batch_job/src/vine_factory.c
+++ b/batch_job/src/vine_factory.c
@@ -494,6 +494,10 @@ static int submit_worker( struct batch_queue *queue )
 	
 	batch_job_set_command(task,cmd);
 	batch_job_add_input_file(task,worker_command,0);
+	
+	if(resources) {
+		batch_job_set_resources(task, resources);
+	}
 
 	if(password_file) {
 		batch_job_add_input_file(task,"pwfile",0);
diff --git a/batch_job/src/work_queue_factory.c b/batch_job/src/work_queue_factory.c
index c6ac9e3c93..470af070c3 100644
--- a/batch_job/src/work_queue_factory.c
+++ b/batch_job/src/work_queue_factory.c
@@ -492,6 +492,10 @@ static int submit_worker( struct batch_queue *queue )
 
 	batch_job_set_command(task,cmd);
 	
+	if(resources) {
+		batch_job_set_resources(task, resources);
+	}
+	
 	if(!runos_os && !k8s_worker_image) {
 		batch_job_add_input_file(task,worker_command,0);
 	} else {