prep

src/codeflare_sdk/utils/generate_yaml.py

@@ -318,10 +318,10 @@ def _throw_for_no_raycluster(    user_yaml: dict,
     api_instance = client.CustomObjectsApi(api_config_handler())
     try:
         api_instance.list_namespaced_custom_object(
-            group="ray.io",
+            group="kueue.x-k8s.io",
             version="v1",
             namespace=namespace,
-            plural="rayclusters",
+            plural="localqueues",
         )
     except ApiException as e:
         if e.status == 404:
@@ -348,7 +348,7 @@ def write_components(
 
     components = user_yaml.get("spec", "resources")["resources"].get("GenericItems")
      # check if RayCluster CustomResourceDefinition exists if not throw RuntimeError
-    _throw_for_no_raycluster()
+    _throw_for_no_raycluster(user_yaml,output_file_name,namespace,local_queue,labels)
     # if local_queue is None:
     #     print(
     #         "Kueue is not installed or won't be used. The absence of CRDs may lack the necessary functionality."