From a829eb9a515925868f73fee78674a30bfcc76042 Mon Sep 17 00:00:00 2001
From: Yousef Moazzam <yousef.moazzam@diamond.ac.uk>
Date: Wed, 12 Aug 2026 15:18:27 +0100
Subject: [PATCH] Add pod spec patch to multi-page tiff generation step

---
 httomo/templates/cor-sweep.yaml | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/httomo/templates/cor-sweep.yaml b/httomo/templates/cor-sweep.yaml
index 49ba1a9..84e88c6 100644
--- a/httomo/templates/cor-sweep.yaml
+++ b/httomo/templates/cor-sweep.yaml
@@ -76,6 +76,10 @@ spec:
         template: sweep-artifact-script
         arguments:
           parameters:
+          - name: nprocs
+            value: "{{`{{ inputs.parameters.nprocs }}`}}"
+          - name: memory
+            value: "{{`{{ inputs.parameters.memory }}`}}"
           - name: output-dir-path
             value: "{{`{{ inputs.parameters.output }}`}}/{{`{{ inputs.parameters.httomo-outdir-name }}`}}"
     tolerations:
@@ -152,6 +156,8 @@ spec:
     inputs:
       parameters:
       - name: output-dir-path
+      - name: nprocs
+      - name: memory
       - name: output-filename
         value: multi-page-tiff.tif
     script:
@@ -185,6 +191,16 @@ spec:
       hostPath:
         path:  "{{`{{ workflow.parameters.visitdir }}`}}"
         type: Directory
+    podSpecPatch: |
+      containers:
+      - name: main
+        resources:
+          requests:
+            cpu: "{{`{{ inputs.parameters.nprocs }}`}}"
+            memory: "{{`{{ inputs.parameters.memory }}`}}"
+          limits:
+            cpu: "{{`{{ inputs.parameters.nprocs }}`}}"
+            memory: "{{`{{ inputs.parameters.memory }}`}}"
     outputs:
       artifacts:
       - name: sweep-res