Skip to content

Commit 8ba6e54

Browse files
create dashboard route with raycluster (#24)
1 parent f0b5377 commit 8ba6e54

File tree

2 files changed

+25
-0
lines changed

2 files changed

+25
-0
lines changed

Diff for: src/codeflare_sdk/templates/new-template.yaml

+16
Original file line numberDiff line numberDiff line change
@@ -232,3 +232,19 @@ spec:
232232
cpu: "2"
233233
memory: "12G"
234234
nvidia.com/gpu: "1"
235+
- replica: 1
236+
generictemplate:
237+
kind: Route
238+
apiVersion: route.openshift.io/v1
239+
metadata:
240+
name: ray-dashboard-deployment-name
241+
namespace: default
242+
labels:
243+
# allows me to return name of service that Ray operator creates
244+
odh-ray-cluster-service: deployment-name-head-svc
245+
spec:
246+
to:
247+
kind: Service
248+
name: deployment-name-head-svc
249+
port:
250+
targetPort: dashboard

Diff for: src/codeflare_sdk/utils/generate_yaml.py

+9
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,13 @@ def gen_names(name):
1919
else:
2020
return name, name
2121

22+
def update_dashboard_route(route_item, cluster_name):
23+
metadata = route_item.get("generictemplate", {}).get("metadata")
24+
metadata["name"] = f'ray-dashboard-{cluster_name}'
25+
metadata["labels"]["odh-ray-cluster-service"] = f'{cluster_name}-head-svc'
26+
spec = route_item.get("generictemplate", {}).get("spec")
27+
spec["to"]["name"] = f'{cluster_name}-head-svc'
28+
2229
def update_names(yaml, item, appwrapper_name, cluster_name):
2330
metadata = yaml.get("metadata")
2431
metadata["name"] = appwrapper_name
@@ -133,10 +140,12 @@ def generate_appwrapper(name, min_cpu, max_cpu, min_memory, max_memory, gpu, wor
133140
appwrapper_name, cluster_name = gen_names(name)
134141
resources = user_yaml.get("spec","resources")
135142
item = resources["resources"].get("GenericItems")[0]
143+
route_item = resources["resources"].get("GenericItems")[1]
136144
update_names(user_yaml, item, appwrapper_name, cluster_name)
137145
update_labels(user_yaml, instascale, instance_types)
138146
update_custompodresources(item, min_cpu, max_cpu, min_memory, max_memory, gpu, workers)
139147
update_nodes(item, appwrapper_name, min_cpu, max_cpu, min_memory, max_memory, gpu, workers, image, instascale, env)
148+
update_dashboard_route(route_item, cluster_name)
140149
outfile = appwrapper_name + ".yaml"
141150
write_user_appwrapper(user_yaml, outfile)
142151
return outfile

0 commit comments

Comments
 (0)