From ced41abac08efe9c06dcb923f76463764074a31d Mon Sep 17 00:00:00 2001
From: MichaelClifford <mcliffor@redhat.com>
Date: Wed, 21 Jun 2023 16:24:18 -0400
Subject: [PATCH] update base template with priorities

---
 src/codeflare_sdk/templates/base-template.yaml | 4 ++++
 tests/test-case-cmd.yaml                       | 3 ++-
 tests/test-case.yaml                           | 3 ++-
 3 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/codeflare_sdk/templates/base-template.yaml b/src/codeflare_sdk/templates/base-template.yaml
index c99fd105d..3f7c31aa7 100644
--- a/src/codeflare_sdk/templates/base-template.yaml
+++ b/src/codeflare_sdk/templates/base-template.yaml
@@ -8,6 +8,8 @@ metadata:
     orderedinstance: "m4.xlarge_g4dn.xlarge"
 spec:
   priority: 9
+  schedulingSpec:
+    minAvailable: 3
   resources:
     Items: []
     GenericItems:
@@ -112,6 +114,7 @@ spec:
                           operator: In
                           values:
                           - "aw-kuberay"
+                priorityClassName: "default-priority"
                 containers:
                 # The Ray head pod
                 - env:
@@ -182,6 +185,7 @@ spec:
                           operator: In
                           values:
                           - "aw-kuberay"
+                priorityClassName: "default-priority"
                 initContainers:
                 # the env var $RAY_IP is set by the operator if missing, with the value of the head service name
                 - name: init-myservice
diff --git a/tests/test-case-cmd.yaml b/tests/test-case-cmd.yaml
index 0d7c66d4f..b57c2ea32 100644
--- a/tests/test-case-cmd.yaml
+++ b/tests/test-case-cmd.yaml
@@ -88,6 +88,7 @@ spec:
                       cpu: 2
                       memory: 8G
                       nvidia.com/gpu: 0
+                priorityClassName: default-priority
           rayVersion: 2.1.0
           workerGroupSpecs:
           - groupName: small-group-unit-cmd-cluster
@@ -152,7 +153,7 @@ spec:
           to:
             kind: Service
             name: unit-cmd-cluster-head-svc
-      replicas: 1
+      replica: 1
     Items: []
   schedulingSpec:
     minAvailable: 3
diff --git a/tests/test-case.yaml b/tests/test-case.yaml
index cf7de08bf..135a72680 100644
--- a/tests/test-case.yaml
+++ b/tests/test-case.yaml
@@ -99,6 +99,7 @@ spec:
                       cpu: 2
                       memory: 8G
                       nvidia.com/gpu: 0
+                priorityClassName: default-priority
           rayVersion: 2.1.0
           workerGroupSpecs:
           - groupName: small-group-unit-test-cluster
@@ -172,7 +173,7 @@ spec:
           to:
             kind: Service
             name: unit-test-cluster-head-svc
-      replicas: 1
+      replica: 1
     Items: []
   schedulingSpec:
     minAvailable: 3