@@ -51,6 +51,8 @@ func TestMNISTRayClusterSDK(t *testing.T) {
51
51
"requirements.txt" : ReadFile (test , "mnist_pip_requirements.txt" ),
52
52
// MNIST training script
53
53
"mnist.py" : ReadFile (test , "mnist.py" ),
54
+ // codeflare-sdk installation script
55
+ "install-codeflare-sdk.sh" : ReadFile (test , "install-codeflare-sdk.sh" ),
54
56
})
55
57
56
58
// Create RBAC, retrieve token for user with limited rights
@@ -119,7 +121,7 @@ func TestMNISTRayClusterSDK(t *testing.T) {
119
121
{Name : "PYTHONUSERBASE" , Value : "/workdir" },
120
122
{Name : "RAY_IMAGE" , Value : GetRayImage ()},
121
123
},
122
- Command : []string {"/bin/sh" , "-c" , "pip install codeflare-sdk==" + GetCodeFlareSDKVersion () + " && cp /test/* . && python mnist_raycluster_sdk.py" + " " + namespace .Name },
124
+ Command : []string {"/bin/sh" , "-c" , "cp /test/* . && chmod +x install-codeflare-sdk.sh && ./install-codeflare-sdk.sh && python mnist_raycluster_sdk.py" + " " + namespace .Name },
123
125
VolumeMounts : []corev1.VolumeMount {
124
126
{
125
127
Name : "test" ,
@@ -129,8 +131,12 @@ func TestMNISTRayClusterSDK(t *testing.T) {
129
131
Name : "workdir" ,
130
132
MountPath : "/workdir" ,
131
133
},
134
+ {
135
+ Name : "workdir" ,
136
+ MountPath : "/workdir/test_config" ,
137
+ },
132
138
},
133
- WorkingDir : "/workdir" ,
139
+ WorkingDir : "/workdir/test_config " ,
134
140
SecurityContext : & corev1.SecurityContext {
135
141
AllowPrivilegeEscalation : Ptr (false ),
136
142
SeccompProfile : & corev1.SeccompProfile {
0 commit comments