update test-case-cmd.yaml

Author: MichaelClifford

tests/test-case-cmd.yaml

@@ -1,8 +1,8 @@
 apiVersion: mcad.ibm.com/v1beta1
 kind: AppWrapper
 metadata:
-  name: unit-cmd-cluster
-  namespace: default
+  name: mnisttest
+  namespace: opendatahub
 spec:
   priority: 9
   resources:
@@ -18,23 +18,23 @@ spec:
           memory: 8G
           nvidia.com/gpu: 0
       - limits:
-          cpu: 1
-          memory: 2G
-          nvidia.com/gpu: 1
+          cpu: 2
+          memory: 4G
+          nvidia.com/gpu: 0
         replicas: 2
         requests:
-          cpu: 1
-          memory: 2G
-          nvidia.com/gpu: 1
+          cpu: 2
+          memory: 4G
+          nvidia.com/gpu: 0
       generictemplate:
         apiVersion: ray.io/v1alpha1
         kind: RayCluster
         metadata:
           labels:
-            appwrapper.mcad.ibm.com: unit-cmd-cluster
+            appwrapper.mcad.ibm.com: mnisttest
             controller-tools.k8s.io: '1.0'
-          name: unit-cmd-cluster
-          namespace: default
+          name: mnisttest
+          namespace: opendatahub
         spec:
           autoscalerOptions:
             idleTimeoutSeconds: 60
@@ -62,7 +62,7 @@ spec:
                     valueFrom:
                       fieldRef:
                         fieldPath: status.podIP
-                  image: rayproject/ray:latest
+                  image: ghcr.io/foundation-model-stack/base:ray2.1.0-py38-gpu-pytorch1.12.0cu116-20221213-193103
                   imagePullPolicy: Always
                   lifecycle:
                     preStop:
@@ -90,12 +90,12 @@ spec:
                       nvidia.com/gpu: 0
           rayVersion: 1.12.0
           workerGroupSpecs:
-          - groupName: small-group-unit-cmd-cluster
+          - groupName: small-group-mnisttest
             maxReplicas: 2
             minReplicas: 2
             rayStartParams:
               block: 'true'
-              num-gpus: '1'
+              num-gpus: '0'
             replicas: 2
             template:
               metadata:
@@ -110,7 +110,7 @@ spec:
                     valueFrom:
                       fieldRef:
                         fieldPath: status.podIP
-                  image: rayproject/ray:latest
+                  image: ghcr.io/foundation-model-stack/base:ray2.1.0-py38-gpu-pytorch1.12.0cu116-20221213-193103
                   lifecycle:
                     preStop:
                       exec:
@@ -121,13 +121,13 @@ spec:
                   name: machine-learning
                   resources:
                     limits:
-                      cpu: 1
-                      memory: 2G
-                      nvidia.com/gpu: 1
+                      cpu: 2
+                      memory: 4G
+                      nvidia.com/gpu: 0
                     requests:
-                      cpu: 1
-                      memory: 2G
-                      nvidia.com/gpu: 1
+                      cpu: 2
+                      memory: 4G
+                      nvidia.com/gpu: 0
                 initContainers:
                 - command:
                   - sh
@@ -142,14 +142,14 @@ spec:
         kind: Route
         metadata:
           labels:
-            odh-ray-cluster-service: unit-cmd-cluster-head-svc
-          name: ray-dashboard-unit-cmd-cluster
-          namespace: default
+            odh-ray-cluster-service: mnisttest-head-svc
+          name: ray-dashboard-mnisttest
+          namespace: opendatahub
         spec:
           port:
             targetPort: dashboard
           to:
             kind: Service
-            name: unit-cmd-cluster-head-svc
+            name: mnisttest-head-svc
       replica: 1
     Items: []