Add label to RayCluster if owned by AppWrapper

Author: ChristianZaccaria

src/codeflare_sdk/ray/cluster/build_ray_cluster.py

@@ -182,7 +182,7 @@ def build_ray_cluster(cluster: "codeflare_sdk.ray.cluster.Cluster"):
 # Metadata related functions
 def get_metadata(cluster: "codeflare_sdk.ray.cluster.Cluster"):
     """
-    The get_metadata() function builds and returns a V1ObjectMeta Object using cluster configurtation parameters
+    The get_metadata() function builds and returns a V1ObjectMeta Object using cluster configuration parameters
     """
     object_meta = V1ObjectMeta(
         name=cluster.config.name,
@@ -207,6 +207,9 @@ def get_labels(cluster: "codeflare_sdk.ray.cluster.Cluster"):
     if cluster.config.labels != {}:
         labels.update(cluster.config.labels)
 
+    if cluster.config.appwrapper is True:
+        labels.update({"resource.owner": "appwrapper"})
+
     if cluster.config.appwrapper is False:
         add_queue_label(cluster, labels)
 

tests/test_cluster_yamls/appwrapper/unit-test-all-params.yaml

@@ -17,6 +17,7 @@ spec:
           controller-tools.k8s.io: '1.0'
           key1: value1
           key2: value2
+          resource.owner: appwrapper
         name: aw-all-params
         namespace: ns
       spec:

tests/test_cluster_yamls/kueue/aw_kueue.yaml

@@ -13,6 +13,7 @@ spec:
       metadata:
         labels:
           controller-tools.k8s.io: '1.0'
+          resource.owner: appwrapper
         name: unit-test-aw-kueue
         namespace: ns
       spec:

tests/test_cluster_yamls/kueue/ray_cluster_kueue.yaml

@@ -13,6 +13,7 @@ spec:
       metadata:
         labels:
           controller-tools.k8s.io: '1.0'
+          resource.owner: appwrapper
         name: unit-test-cluster-kueue
         namespace: ns
       spec:

tests/test_cluster_yamls/ray/default-appwrapper.yaml

@@ -11,6 +11,7 @@ spec:
       metadata:
         labels:
           controller-tools.k8s.io: '1.0'
+          resource.owner: appwrapper
         name: default-appwrapper
         namespace: ns
       spec: