Skip to content

Commit 16d7ad2

Browse files
committed
eliminate dependency on workload.codeflare.dev/appwrapper label
1 parent b313b15 commit 16d7ad2

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

@@ -228,16 +227,9 @@ def write_components(
228227
with open(output_file_name, "a") as outfile:
229228
for component in components:
230229
if "generictemplate" in component:
231-
if (
232-
"workload.codeflare.dev/appwrapper"
233-
in component["generictemplate"]["metadata"]["labels"]
234-
):
235-
del component["generictemplate"]["metadata"]["labels"][
236-
"workload.codeflare.dev/appwrapper"
237-
]
238-
labels = component["generictemplate"]["metadata"]["labels"]
239-
labels.update({"kueue.x-k8s.io/queue-name": lq_name})
240-
labels.update(cluster_labels)
230+
labels = component["generictemplate"]["metadata"]["labels"]
231+
labels.update({"kueue.x-k8s.io/queue-name": lq_name})
232+
labels.update(cluster_labels)
241233
outfile.write("---\n")
242234
yaml.dump(
243235
component["generictemplate"], outfile, default_flow_style=False
@@ -258,16 +250,9 @@ def load_components(
258250
cluster_labels = labels
259251
for component in components:
260252
if "generictemplate" in component:
261-
if (
262-
"workload.codeflare.dev/appwrapper"
263-
in component["generictemplate"]["metadata"]["labels"]
264-
):
265-
del component["generictemplate"]["metadata"]["labels"][
266-
"workload.codeflare.dev/appwrapper"
267-
]
268-
labels = component["generictemplate"]["metadata"]["labels"]
269-
labels.update({"kueue.x-k8s.io/queue-name": lq_name})
270-
labels.update(cluster_labels)
253+
labels = component["generictemplate"]["metadata"]["labels"]
254+
labels.update({"kueue.x-k8s.io/queue-name": lq_name})
255+
labels.update(cluster_labels)
271256
component_list.append(component["generictemplate"])
272257

273258
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
@@ -968,7 +968,6 @@ def get_ray_obj(group, version, namespace, plural, cls=None):
968968
"labels": {
969969
"controller-tools.k8s.io": "1.0",
970970
"resourceName": "quicktest",
971-
"workload.codeflare.dev/appwrapper": "quicktest",
972971
"orderedinstance": "m4.xlarge_g4dn.xlarge",
973972
},
974973
"name": "quicktest",
@@ -1275,7 +1274,6 @@ def get_ray_obj(group, version, namespace, plural, cls=None):
12751274
"creationTimestamp": "2023-02-22T16:26:07Z",
12761275
"generation": 1,
12771276
"labels": {
1278-
"workload.codeflare.dev/appwrapper": "quicktest2",
12791277
"controller-tools.k8s.io": "1.0",
12801278
"resourceName": "quicktest2",
12811279
"orderedinstance": "m4.xlarge_g4dn.xlarge",
@@ -1466,7 +1464,6 @@ def get_aw_obj(group, version, namespace, plural):
14661464
"kind": "RayCluster",
14671465
"metadata": {
14681466
"labels": {
1469-
"workload.codeflare.dev/appwrapper": "quicktest1",
14701467
"controller-tools.k8s.io": "1.0",
14711468
},
14721469
"name": "quicktest1",
@@ -1670,7 +1667,6 @@ def get_aw_obj(group, version, namespace, plural):
16701667
"kind": "RayCluster",
16711668
"metadata": {
16721669
"labels": {
1673-
"workload.codeflare.dev/appwrapper": "quicktest2",
16741670
"controller-tools.k8s.io": "1.0",
16751671
},
16761672
"name": "quicktest2",

0 commit comments

Comments
 (0)