@@ -27,22 +27,15 @@ func runFailedPyTorchJobTest(t *testing.T, image string) {
27
27
// Create a namespace
28
28
namespace := test .NewTestNamespace ()
29
29
30
- // Create a ConfigMap with training dataset and configuration
31
- configData := map [string ][]byte {
32
- "config.json" : ReadFile (test , "config.json" ),
33
- "twitter_complaints_small.json" : ReadFile (test , "twitter_complaints_small.json" ),
34
- }
35
- config := CreateConfigMap (test , namespace .Name , configData )
36
-
37
30
// Create training PyTorch job
38
- tuningJob := createFailedPyTorchJob (test , namespace .Name , * config , image )
31
+ tuningJob := createFailedPyTorchJob (test , namespace .Name , image )
39
32
40
33
// Make sure the PyTorch job is failed
41
- test .Eventually (PyTorchJob (test , namespace .Name , tuningJob .Name ), TestTimeoutLong ).
34
+ test .Eventually (PyTorchJob (test , namespace .Name , tuningJob .Name ), TestTimeoutDouble ).
42
35
Should (WithTransform (PyTorchJobConditionFailed , Equal (corev1 .ConditionTrue )))
43
36
}
44
37
45
- func createFailedPyTorchJob (test Test , namespace string , config corev1. ConfigMap , baseImage string ) * kftov1.PyTorchJob {
38
+ func createFailedPyTorchJob (test Test , namespace string , baseImage string ) * kftov1.PyTorchJob {
46
39
tuningJob := & kftov1.PyTorchJob {
47
40
TypeMeta : metav1.TypeMeta {
48
41
APIVersion : corev1 .SchemeGroupVersion .String (),
@@ -80,18 +73,6 @@ func createFailedPyTorchJob(test Test, namespace string, config corev1.ConfigMap
80
73
},
81
74
},
82
75
},
83
- Volumes : []corev1.Volume {
84
- {
85
- Name : "config-volume" ,
86
- VolumeSource : corev1.VolumeSource {
87
- ConfigMap : & corev1.ConfigMapVolumeSource {
88
- LocalObjectReference : corev1.LocalObjectReference {
89
- Name : config .Name ,
90
- },
91
- },
92
- },
93
- },
94
- },
95
76
},
96
77
},
97
78
},
0 commit comments