Skip to content

Commit 51f1ed0

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

File tree

1 file changed

+8
-11
lines changed

1 file changed

+8
-11
lines changed

src/codeflare_sdk/utils/generate_yaml.py

Lines changed: 8 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -344,15 +344,16 @@ def write_components(
344344
components = user_yaml.get("spec", "resources")["resources"].get("GenericItems")
345345
open(output_file_name, "w").close()
346346
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-
)
351347
cluster_labels = labels
352348
if not local_queue_exists(namespace, lq_name):
353-
raise ValueError(
354-
"local_queue provided does not exist or is not in this namespace. Please provide the correct local_queue name in Cluster Configuration"
355-
)
349+
if lq_name is None:
350+
print(
351+
"Kueue is not installed or won't be used. The absence of CRDs may lack the necessary functionality."
352+
)
353+
else:
354+
raise ValueError(
355+
"local_queue provided does not exist or is not in this namespace. Please provide the correct local_queue name in Cluster Configuration"
356+
)
356357
with open(output_file_name, "a") as outfile:
357358
for component in components:
358359
if "generictemplate" in component:
@@ -383,10 +384,6 @@ def load_components(
383384
component_list = []
384385
components = user_yaml.get("spec", "resources")["resources"].get("GenericItems")
385386
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-
)
390387
cluster_labels = labels
391388
if not local_queue_exists(namespace, lq_name):
392389
raise ValueError(

0 commit comments

Comments
 (0)