@@ -51,6 +51,8 @@ func TestMNISTRayClusterSDK(t *testing.T) {
5151 "requirements.txt" : ReadFile (test , "mnist_pip_requirements.txt" ),
5252 // MNIST training script
5353 "mnist.py" : ReadFile (test , "mnist.py" ),
54+ // codeflare-sdk installation script
55+ "install-codeflare-sdk.sh" : ReadFile (test , "install-codeflare-sdk.sh" ),
5456 })
5557
5658 // Create RBAC, retrieve token for user with limited rights
@@ -119,12 +121,16 @@ func TestMNISTRayClusterSDK(t *testing.T) {
119121 {Name : "PYTHONUSERBASE" , Value : "/workdir" },
120122 {Name : "RAY_IMAGE" , Value : GetRayImage ()},
121123 },
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 },
123125 VolumeMounts : []corev1.VolumeMount {
124126 {
125127 Name : "test" ,
126128 MountPath : "/test" ,
127129 },
130+ {
131+ Name : "codeflare-sdk" ,
132+ MountPath : "/codeflare-sdk" ,
133+ },
128134 {
129135 Name : "workdir" ,
130136 MountPath : "/workdir" ,
@@ -154,6 +160,12 @@ func TestMNISTRayClusterSDK(t *testing.T) {
154160 },
155161 },
156162 },
163+ {
164+ Name : "codeflare-sdk" ,
165+ VolumeSource : corev1.VolumeSource {
166+ EmptyDir : & corev1.EmptyDirVolumeSource {},
167+ },
168+ },
157169 {
158170 Name : "workdir" ,
159171 VolumeSource : corev1.VolumeSource {
0 commit comments