@@ -19,6 +19,13 @@ def gen_names(name):
19
19
else :
20
20
return name , name
21
21
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
+
22
29
def update_names (yaml , item , appwrapper_name , cluster_name ):
23
30
metadata = yaml .get ("metadata" )
24
31
metadata ["name" ] = appwrapper_name
@@ -133,10 +140,12 @@ def generate_appwrapper(name, min_cpu, max_cpu, min_memory, max_memory, gpu, wor
133
140
appwrapper_name , cluster_name = gen_names (name )
134
141
resources = user_yaml .get ("spec" ,"resources" )
135
142
item = resources ["resources" ].get ("GenericItems" )[0 ]
143
+ route_item = resources ["resources" ].get ("GenericItems" )[1 ]
136
144
update_names (user_yaml , item , appwrapper_name , cluster_name )
137
145
update_labels (user_yaml , instascale , instance_types )
138
146
update_custompodresources (item , min_cpu , max_cpu , min_memory , max_memory , gpu , workers )
139
147
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 )
140
149
outfile = appwrapper_name + ".yaml"
141
150
write_user_appwrapper (user_yaml , outfile )
142
151
return outfile
0 commit comments