From 2c0dfe241eda18737c9ff213c38ce226e8cb7f59 Mon Sep 17 00:00:00 2001
From: frederick <frederick.allars@diamond.ac.uk>
Date: Thu, 23 Jul 2026 11:02:53 +0100
Subject: [PATCH] adding tolerations, volumes and updating artifacts

---
 e02/templates/Notebook.yaml | 24 +++++++++++++++++-------
 1 file changed, 17 insertions(+), 7 deletions(-)

diff --git a/e02/templates/Notebook.yaml b/e02/templates/Notebook.yaml
index e82a625..16d0d0a 100644
--- a/e02/templates/Notebook.yaml
+++ b/e02/templates/Notebook.yaml
@@ -1,14 +1,14 @@
 apiVersion: argoproj.io/v1alpha1
 kind: ClusterWorkflowTemplate
 metadata:
-  name: calibrate-ptycho
+  name: epsic-calibrate-ptycho
   labels:
     workflows.diamond.ac.uk/science-group-imaging: "true"
   annotations:
+    workflows.diamond.ac.uk/repository: "https://github.com/DiamondLightSource/imaging-workflows"
     workflows.argoproj.io/title: Calibrate ptycho STEM data
     workflows.argoproj.io/description: |
       A utility which takes in a 4D STEM data set and outs the approximate real space pixel size, measured camera length and scan rotation, based off dpc
-    workflows.diamond.ac.uk/repository: "https://github.com/DiamondLightSource/imaging-workflows"
     workflows.diamond.ac.uk/parameter-schema: |
       {{- .Files.Get "schema/notebook_Schema.json" | nindent 6 }}
     workflows.diamond.ac.uk/ui-schema: |
@@ -99,11 +99,7 @@ spec:
       outputs:
         artifacts:
         - name: phase-output
-          path: "{{`{{ inputs.parameters.outpath }}`}}/phase-output.html"
-          archive:
-            none: {}
-        - name: notebook-log
-          path: "{{`{{ inputs.parameters.outpath }}`}}/papermill.log"
+          path: "{{`{{workflow.parameters.visitdir}}`}}/Calibrate-notebook.html"
           archive:
             none: {}
       podSpecPatch: |
@@ -116,5 +112,19 @@ spec:
             limits:
               cpu: 10
               memory: 50Gi
+      tolerations:
+      - key: nodegroup
+        operator: Equal
+        value: workflows
+        effect: NoSchedule
+      volumes:
+      - name: session
+        hostPath:
+          path: "{{`{{ workflow.parameters.visitdir }}`}}"
+          type: Directory
+      - name: software
+        hostPath:
+          path: /dls_sw/e02/medipix_mask/
+          type: Directory
 
         
\ No newline at end of file