30
30
from os import urandom
31
31
from base64 import b64encode
32
32
from urllib3 .util import parse_url
33
- from kubernetes .client .rest import ApiException
34
33
35
34
36
35
def read_template (template ):
@@ -308,32 +307,6 @@ def get_default_kueue_name(namespace: str):
308
307
"Default Local Queue with kueue.x-k8s.io/default-queue: true annotation not found please create a default Local Queue or provide the local_queue name in Cluster Configuration"
309
308
)
310
309
311
- # test function
312
-
313
- # def _throw_for_no_raycluster( user_yaml: dict,
314
- # output_file_name: str,
315
- # namespace: str,
316
- # local_queue: Optional[str],
317
- # labels: dict,):
318
- # api_instance = client.CustomObjectsApi(api_config_handler())
319
- # print('hello')
320
- # # try:
321
- # # api_instance.list_namespaced_custom_object(
322
- # # group="kueue.x-k8s.io",
323
- # # version="v1",
324
- # # namespace=namespace,
325
- # # plural="localqueues",
326
- # # )
327
- # # except ApiException as e:
328
- # # if e.status == 404:
329
- # # raise RuntimeError(
330
- # # "RayCluster CustomResourceDefinition unavailable contact your administrator."
331
- # # )
332
- # # else:
333
- # # raise RuntimeError(
334
- # # "Failed to get RayCluster CustomResourceDefinition: " + str(e)
335
- # # )
336
-
337
310
338
311
def local_queue_exists (namespace : str , local_queue_name : str ):
339
312
# get all local queues in the namespace
@@ -368,12 +341,6 @@ def write_components(
368
341
os .makedirs (directory_path )
369
342
370
343
components = user_yaml .get ("spec" , "resources" )["resources" ].get ("GenericItems" )
371
- # check if RayCluster CustomResourceDefinition exists if not throw RuntimeError
372
- # _throw_for_no_raycluster()
373
- # if local_queue is None:
374
- # print(
375
- # "Kueue is not installed or won't be used. The absence of CRDs may lack the necessary functionality."
376
- # )
377
344
open (output_file_name , "w" ).close ()
378
345
lq_name = local_queue or get_default_kueue_name (namespace )
379
346
cluster_labels = labels
@@ -400,31 +367,22 @@ def write_components(
400
367
)
401
368
print (f"Written to: { output_file_name } " )
402
369
370
+
403
371
def load_components (
404
372
user_yaml : dict ,
405
373
name : str ,
406
374
namespace : str ,
407
375
local_queue : Optional [str ],
408
- labels : dict
376
+ labels : dict ,
409
377
):
410
378
component_list = []
411
379
components = user_yaml .get ("spec" , "resources" )["resources" ].get ("GenericItems" )
412
380
lq_name = local_queue or get_default_kueue_name (namespace )
413
381
cluster_labels = labels
414
- < << << << HEAD
415
382
if not local_queue_exists (namespace , lq_name ):
416
383
raise ValueError (
417
384
"local_queue provided does not exist or is not in this namespace. Please provide the correct local_queue name in Cluster Configuration"
418
385
)
419
- == == == =
420
-
421
- # _throw_for_no_raycluster()
422
-
423
- # if local_queue is None:
424
- # print(
425
- # "Kueue is not installed or won't be used. The absence of CRDs may lack the necessary functionality."
426
- # )
427
- >> >> >> > 1 d38c8d (prep )
428
386
for component in components :
429
387
if "generictemplate" in component :
430
388
if (
0 commit comments