Skip to content

Commit 1666e8a

Browse files
committed
Add warning if Kueue CRDs not installed on cluster
1 parent 2a43d9e commit 1666e8a

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

src/codeflare_sdk/utils/generate_yaml.py

Lines changed: 9 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@
3030
from os import urandom
3131
from base64 import b64encode
3232
from urllib3.util import parse_url
33+
from kubernetes.client.rest import ApiException
3334

3435

3536
def read_template(template):
@@ -343,6 +344,10 @@ def write_components(
343344
components = user_yaml.get("spec", "resources")["resources"].get("GenericItems")
344345
open(output_file_name, "w").close()
345346
lq_name = local_queue or get_default_kueue_name(namespace)
347+
if lq_name is None:
348+
print(
349+
"Kueue is not installed or won't be used. The absence of CRDs may lack the necessary functionality."
350+
)
346351
cluster_labels = labels
347352
if not local_queue_exists(namespace, lq_name):
348353
raise ValueError(
@@ -378,6 +383,10 @@ def load_components(
378383
component_list = []
379384
components = user_yaml.get("spec", "resources")["resources"].get("GenericItems")
380385
lq_name = local_queue or get_default_kueue_name(namespace)
386+
if lq_name is None:
387+
print(
388+
"Kueue is not installed or won't be used. The absence of CRDs may lack the necessary functionality."
389+
)
381390
cluster_labels = labels
382391
if not local_queue_exists(namespace, lq_name):
383392
raise ValueError(

0 commit comments

Comments
 (0)