fix: pipeline run

redhat-na-ssa · Oct 3, 2024 · 3c2a083 · 3c2a083
1 parent 8ffb8db
commit 3c2a083
Showing 1 changed file with 8 additions and 9 deletions.
diff --git a/scripts/pipeline-run.yaml b/scripts/pipeline-run.yaml
@@ -7,15 +7,14 @@ metadata:
   labels:
     tekton.dev/pipeline: train-model
 spec:
-  taskRunTemplate:
-    podTemplate:
-      tolerations:
-        - key: nvidia-gpu-only
-          operator: Exists
-          effect: NoSchedule
-        - key: nvidia.com/gpu
-          operator: Exists
-          effect: NoSchedule
+  podTemplate:
+    tolerations:
+      - key: nvidia-gpu-only
+        operator: Exists
+        effect: NoSchedule
+      - key: nvidia.com/gpu
+        operator: Exists
+        effect: NoSchedule
   params:
     - name: BATCH_SIZE
       value: '-1'