@@ -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,7 +121,7 @@ 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" ,
@@ -129,8 +131,12 @@ func TestMNISTRayClusterSDK(t *testing.T) {
129131 Name : "workdir" ,
130132 MountPath : "/workdir" ,
131133 },
134+ {
135+ Name : "workdir" ,
136+ MountPath : "/workdir/test_config" ,
137+ },
132138 },
133- WorkingDir : "/workdir" ,
139+ WorkingDir : "/workdir/test_config " ,
134140 SecurityContext : & corev1.SecurityContext {
135141 AllowPrivilegeEscalation : Ptr (false ),
136142 SeccompProfile : & corev1.SeccompProfile {
0 commit comments