Fix Training operator upgrade test

opendatahub-io · Nov 29, 2024 · 5d41c7a · 5d41c7a
1 parent 53c2eb4
commit 5d41c7a
Showing 1 changed file with 10 additions and 0 deletions.
diff --git 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{},
+									},
+								},
 							},
 						},
 					},