From e82b3e4d1874fae544ba79bba6fd9c0f3efd1536 Mon Sep 17 00:00:00 2001
From: Venkat <venkatajagannath.yellapragada@astronomer.io>
Date: Tue, 6 Aug 2024 17:22:26 -0400
Subject: [PATCH] yaml files updated

---
 example_dags/scripts/k8-gpu.yaml      | 15 ---------------
 example_dags/scripts/k8.yaml          | 15 ---------------
 example_dags/scripts/ray-gpu.yaml     |  4 ++--
 example_dags/scripts/ray-service.yaml | 27 ---------------------------
 example_dags/scripts/ray.yaml         | 11 ++---------
 5 files changed, 4 insertions(+), 68 deletions(-)
 delete mode 100644 example_dags/scripts/k8-gpu.yaml
 delete mode 100644 example_dags/scripts/k8.yaml
 delete mode 100644 example_dags/scripts/ray-service.yaml

diff --git a/example_dags/scripts/k8-gpu.yaml b/example_dags/scripts/k8-gpu.yaml
deleted file mode 100644
index c823f78..0000000
--- a/example_dags/scripts/k8-gpu.yaml
+++ /dev/null
@@ -1,15 +0,0 @@
-apiVersion: eksctl.io/v1alpha5
-kind: ClusterConfig
-
-metadata:
-  name: RayCluster
-  region: us-east-2
-
-managedNodeGroups:
-  - name: ng-1-gpu
-    instanceType: g4dn.2xlarge
-    desiredCapacity: 2
-    minSize: 1
-    maxSize: 3
-    ssh:
-      allow: false
diff --git a/example_dags/scripts/k8.yaml b/example_dags/scripts/k8.yaml
deleted file mode 100644
index fd68b5a..0000000
--- a/example_dags/scripts/k8.yaml
+++ /dev/null
@@ -1,15 +0,0 @@
-apiVersion: eksctl.io/v1alpha5
-kind: ClusterConfig
-
-metadata:
-  name: RayCluster
-  region: us-east-2
-
-managedNodeGroups:
-  - name: ng-1
-    instanceType: m5.2xlarge
-    desiredCapacity: 2
-    minSize: 1
-    maxSize: 3
-    ssh:
-      allow: false # This will use the default SSH key on your system (optional)
diff --git a/example_dags/scripts/ray-gpu.yaml b/example_dags/scripts/ray-gpu.yaml
index da7b596..c79860d 100644
--- a/example_dags/scripts/ray-gpu.yaml
+++ b/example_dags/scripts/ray-gpu.yaml
@@ -1,4 +1,4 @@
-apiVersion: ray.io/v1alpha1
+apiVersion: ray.io/v1
 kind: RayCluster
 metadata:
   name: raycluster-complete
@@ -6,7 +6,7 @@ spec:
   rayVersion: "2.10.0"
   enableInTreeAutoscaling: true
   headGroupSpec:
-    serviceType: ClusterIP
+    serviceType: LoadBalancer
     rayStartParams:
       dashboard-host: "0.0.0.0"
       block: "true"
diff --git a/example_dags/scripts/ray-service.yaml b/example_dags/scripts/ray-service.yaml
deleted file mode 100644
index 9e6490c..0000000
--- a/example_dags/scripts/ray-service.yaml
+++ /dev/null
@@ -1,27 +0,0 @@
-apiVersion: v1
-kind: Service
-metadata:
-  name: ray-service
-  namespace: ray
-spec:
-  selector:
-    ray.io/cluster: raycluster-complete
-    ray.io/node-type: head
-  ports:
-  - name: dashboard
-    protocol: TCP
-    port: 8265
-    targetPort: 8265
-  - name: client
-    protocol: TCP
-    port: 10001
-    targetPort: 10001
-  - name: gcs
-    protocol: TCP
-    port: 6379
-    targetPort: 6379
-  - name: serve
-    protocol: TCP
-    port: 8000
-    targetPort: 8000
-  type: LoadBalancer
diff --git a/example_dags/scripts/ray.yaml b/example_dags/scripts/ray.yaml
index 01e1e33..1eb8e70 100644
--- a/example_dags/scripts/ray.yaml
+++ b/example_dags/scripts/ray.yaml
@@ -1,4 +1,4 @@
-apiVersion: ray.io/v1alpha1
+apiVersion: ray.io/v1
 kind: RayCluster
 metadata:
   name: raycluster-complete
@@ -6,7 +6,7 @@ spec:
   rayVersion: "2.10.0"
   enableInTreeAutoscaling: true
   headGroupSpec:
-    serviceType: ClusterIP
+    serviceType: LoadBalancer
     rayStartParams:
       dashboard-host: "0.0.0.0"
       block: "true"
@@ -18,13 +18,6 @@ spec:
         containers:
         - name: ray-head
           image: rayproject/ray-ml:latest
-          env:  # Environment variables section starts here
-          - name: RAY_GRAFANA_IFRAME_HOST
-            value: "http://127.0.0.1:3000"
-          - name: RAY_GRAFANA_HOST
-            value: "http://prometheus-grafana.prometheus-system.svc:80"
-          - name: RAY_PROMETHEUS_HOST
-            value: "http://prometheus-kube-prometheus-prometheus.prometheus-system.svc:9090"
           resources:
             limits:
               cpu: 4