Added optional kueue parameter

Author: Bobbins228

src/codeflare_sdk/cluster/cluster.py

@@ -152,6 +152,7 @@ def create_app_wrapper(self):
         write_to_file = self.config.write_to_file
         local_queue = self.config.local_queue
         labels = self.config.labels
+        kueue = self.config.kueue
         return generate_appwrapper(
             name=name,
             namespace=namespace,
@@ -172,6 +173,7 @@ def create_app_wrapper(self):
             write_to_file=write_to_file,
             local_queue=local_queue,
             labels=labels,
+            kueue=kueue,
         )
 
     # creates a new cluster with the provided or default spec

src/codeflare_sdk/cluster/config.py

@@ -60,6 +60,7 @@ class ClusterConfiguration:
     write_to_file: bool = False
     verify_tls: bool = True
     labels: dict = field(default_factory=dict)
+    kueue: bool = True
 
     def __post_init__(self):
         if not self.verify_tls:

src/codeflare_sdk/utils/generate_yaml.py

@@ -294,6 +294,7 @@ def generate_appwrapper(
     write_to_file: bool,
     local_queue: Optional[str],
     labels,
+    kueue: bool,
 ):
     cluster_yaml = read_template(template)
     appwrapper_name, cluster_name = gen_names(name)
@@ -323,7 +324,8 @@ def generate_appwrapper(
         else cluster_yaml
     )
 
-    add_queue_label(user_yaml, namespace, local_queue)
+    if kueue:
+        add_queue_label(user_yaml, namespace, local_queue)
 
     if write_to_file:
         directory_path = os.path.expanduser("~/.codeflare/resources/")