@@ -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,12 +121,16 @@ 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" ,
126
128
MountPath : "/test" ,
127
129
},
130
+ {
131
+ Name : "codeflare-sdk" ,
132
+ MountPath : "/codeflare-sdk" ,
133
+ },
128
134
{
129
135
Name : "workdir" ,
130
136
MountPath : "/workdir" ,
@@ -154,6 +160,12 @@ func TestMNISTRayClusterSDK(t *testing.T) {
154
160
},
155
161
},
156
162
},
163
+ {
164
+ Name : "codeflare-sdk" ,
165
+ VolumeSource : corev1.VolumeSource {
166
+ EmptyDir : & corev1.EmptyDirVolumeSource {},
167
+ },
168
+ },
157
169
{
158
170
Name : "workdir" ,
159
171
VolumeSource : corev1.VolumeSource {
0 commit comments