Skip to content

Commit 7eacf77

Browse files
committed
eliminate dependency on workload.codeflare.dev/appwrapper label
1 parent 40e761d commit 7eacf77

File tree

6 files changed

+6
-29
lines changed

6 files changed

+6
-29
lines changed

src/codeflare_sdk/templates/base-template.yaml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,6 @@ spec:
1616
kind: RayCluster
1717
metadata:
1818
labels:
19-
workload.codeflare.dev/appwrapper: "aw-kuberay"
2019
controller-tools.k8s.io: "1.0"
2120
# A unique identifier for the head node and workers of this cluster.
2221
name: kuberay-cluster

src/codeflare_sdk/utils/generate_yaml.py

Lines changed: 6 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -82,7 +82,6 @@ def update_names(yaml, item, appwrapper_name, cluster_name, namespace):
8282
metadata["name"] = appwrapper_name
8383
metadata["namespace"] = namespace
8484
lower_meta = item.get("generictemplate", {}).get("metadata")
85-
lower_meta["labels"]["workload.codeflare.dev/appwrapper"] = appwrapper_name
8685
lower_meta["name"] = cluster_name
8786
lower_meta["namespace"] = namespace
8887

@@ -252,16 +251,9 @@ def write_components(
252251
with open(output_file_name, "a") as outfile:
253252
for component in components:
254253
if "generictemplate" in component:
255-
if (
256-
"workload.codeflare.dev/appwrapper"
257-
in component["generictemplate"]["metadata"]["labels"]
258-
):
259-
del component["generictemplate"]["metadata"]["labels"][
260-
"workload.codeflare.dev/appwrapper"
261-
]
262-
labels = component["generictemplate"]["metadata"]["labels"]
263-
labels.update({"kueue.x-k8s.io/queue-name": lq_name})
264-
labels.update(cluster_labels)
254+
labels = component["generictemplate"]["metadata"]["labels"]
255+
labels.update({"kueue.x-k8s.io/queue-name": lq_name})
256+
labels.update(cluster_labels)
265257
outfile.write("---\n")
266258
yaml.dump(
267259
component["generictemplate"], outfile, default_flow_style=False
@@ -286,16 +278,9 @@ def load_components(
286278
)
287279
for component in components:
288280
if "generictemplate" in component:
289-
if (
290-
"workload.codeflare.dev/appwrapper"
291-
in component["generictemplate"]["metadata"]["labels"]
292-
):
293-
del component["generictemplate"]["metadata"]["labels"][
294-
"workload.codeflare.dev/appwrapper"
295-
]
296-
labels = component["generictemplate"]["metadata"]["labels"]
297-
labels.update({"kueue.x-k8s.io/queue-name": lq_name})
298-
labels.update(cluster_labels)
281+
labels = component["generictemplate"]["metadata"]["labels"]
282+
labels.update({"kueue.x-k8s.io/queue-name": lq_name})
283+
labels.update(cluster_labels)
299284
component_list.append(component["generictemplate"])
300285

301286
resources = "---\n" + "---\n".join(

tests/test-case-bad.yaml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,6 @@ spec:
1313
kind: RayCluster
1414
metadata:
1515
labels:
16-
workload.codeflare.dev/appwrapper: unit-test-cluster
1716
controller-tools.k8s.io: '1.0'
1817
name: unit-test-cluster
1918
namespace: ns

tests/test-case.yaml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,6 @@ spec:
1212
metadata:
1313
labels:
1414
controller-tools.k8s.io: '1.0'
15-
workload.codeflare.dev/appwrapper: unit-test-cluster
1615
name: unit-test-cluster
1716
namespace: ns
1817
spec:

tests/test-default-appwrapper.yaml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,6 @@ spec:
1212
metadata:
1313
labels:
1414
controller-tools.k8s.io: '1.0'
15-
workload.codeflare.dev/appwrapper: unit-test-default-cluster
1615
name: unit-test-default-cluster
1716
namespace: opendatahub
1817
spec:

tests/unit_test.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -994,7 +994,6 @@ def get_ray_obj(group, version, namespace, plural, cls=None):
994994
"labels": {
995995
"controller-tools.k8s.io": "1.0",
996996
"resourceName": "quicktest",
997-
"workload.codeflare.dev/appwrapper": "quicktest",
998997
"orderedinstance": "m4.xlarge_g4dn.xlarge",
999998
"kueue.x-k8s.io/queue-name": "team-a-queue",
1000999
},
@@ -1302,7 +1301,6 @@ def get_ray_obj(group, version, namespace, plural, cls=None):
13021301
"creationTimestamp": "2023-02-22T16:26:07Z",
13031302
"generation": 1,
13041303
"labels": {
1305-
"workload.codeflare.dev/appwrapper": "quicktest2",
13061304
"controller-tools.k8s.io": "1.0",
13071305
"resourceName": "quicktest2",
13081306
"orderedinstance": "m4.xlarge_g4dn.xlarge",
@@ -1493,7 +1491,6 @@ def get_aw_obj(group, version, namespace, plural):
14931491
"kind": "RayCluster",
14941492
"metadata": {
14951493
"labels": {
1496-
"workload.codeflare.dev/appwrapper": "quicktest1",
14971494
"controller-tools.k8s.io": "1.0",
14981495
},
14991496
"name": "quicktest1",
@@ -1697,7 +1694,6 @@ def get_aw_obj(group, version, namespace, plural):
16971694
"kind": "RayCluster",
16981695
"metadata": {
16991696
"labels": {
1700-
"workload.codeflare.dev/appwrapper": "quicktest2",
17011697
"controller-tools.k8s.io": "1.0",
17021698
},
17031699
"name": "quicktest2",

0 commit comments

Comments
 (0)