@@ -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+
2229def 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