From 5d41c7ab1cf0383e5219a157b7584d8467e7370c Mon Sep 17 00:00:00 2001
From: Karel Suta <ksuta@redhat.com>
Date: Fri, 29 Nov 2024 11:13:49 +0100
Subject: [PATCH] Fix Training operator upgrade test

---
 .../upgrade/kfto_kueue_sft_upgrade_training_test.go    | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/tests/kfto/upgrade/kfto_kueue_sft_upgrade_training_test.go b/tests/kfto/upgrade/kfto_kueue_sft_upgrade_training_test.go
index 9ec9101f..98642d26 100644
--- a/tests/kfto/upgrade/kfto_kueue_sft_upgrade_training_test.go
+++ b/tests/kfto/upgrade/kfto_kueue_sft_upgrade_training_test.go
@@ -186,6 +186,10 @@ func createPyTorchJob(test Test, namespace, localQueueName string, config corev1
 											Name:      "tmp-volume",
 											MountPath: "/tmp",
 										},
+										{
+											Name:      "output-volume",
+											MountPath: "/mnt/output",
+										},
 									},
 									Resources: corev1.ResourceRequirements{
 										Requests: corev1.ResourceList{
@@ -226,6 +230,12 @@ func createPyTorchJob(test Test, namespace, localQueueName string, config corev1
 										EmptyDir: &corev1.EmptyDirVolumeSource{},
 									},
 								},
+								{
+									Name: "output-volume",
+									VolumeSource: corev1.VolumeSource{
+										EmptyDir: &corev1.EmptyDirVolumeSource{},
+									},
+								},
 							},
 						},
 					},