diff --git a/samples/wrapped-deployment.yaml b/samples/wrapped-deployment.yaml index 05fa200..a5d0ce1 100644 --- a/samples/wrapped-deployment.yaml +++ b/samples/wrapped-deployment.yaml @@ -2,7 +2,7 @@ apiVersion: workload.codeflare.dev/v1beta2 kind: AppWrapper metadata: name: sample-deployment - annotations: + labels: kueue.x-k8s.io/queue-name: user-queue spec: components: diff --git a/samples/wrapped-failing-job.yaml b/samples/wrapped-failing-job.yaml index 3b8eb59..e69de64 100644 --- a/samples/wrapped-failing-job.yaml +++ b/samples/wrapped-failing-job.yaml @@ -2,8 +2,9 @@ apiVersion: workload.codeflare.dev/v1beta2 kind: AppWrapper metadata: name: sample-failing-job - annotations: + labels: kueue.x-k8s.io/queue-name: user-queue + annotations: workload.codeflare.dev.appwrapper/failureGracePeriodDuration: 10s workload.codeflare.dev.appwrapper/resetPauseDuration: 10s workload.codeflare.dev.appwrapper/retryLimit: "3" diff --git a/samples/wrapped-failing-pod.yaml b/samples/wrapped-failing-pod.yaml index 5148999..acdc450 100644 --- a/samples/wrapped-failing-pod.yaml +++ b/samples/wrapped-failing-pod.yaml @@ -2,8 +2,9 @@ apiVersion: workload.codeflare.dev/v1beta2 kind: AppWrapper metadata: name: sample-failing-pod - annotations: + labels: kueue.x-k8s.io/queue-name: user-queue + annotations: workload.codeflare.dev.appwrapper/failureGracePeriodDuration: 10s workload.codeflare.dev.appwrapper/resetPauseDuration: 10s workload.codeflare.dev.appwrapper/retryLimit: "1" diff --git a/samples/wrapped-job.yaml b/samples/wrapped-job.yaml index a9fdba8..8099b7c 100644 --- a/samples/wrapped-job.yaml +++ b/samples/wrapped-job.yaml @@ -2,8 +2,9 @@ apiVersion: workload.codeflare.dev/v1beta2 kind: AppWrapper metadata: name: sample-job - annotations: + labels: kueue.x-k8s.io/queue-name: user-queue + annotations: workload.codeflare.dev.appwrapper/successTTLDuration: "1m" spec: components: diff --git a/samples/wrapped-pod.yaml b/samples/wrapped-pod.yaml index 90d3538..9f24eb8 100644 --- a/samples/wrapped-pod.yaml +++ b/samples/wrapped-pod.yaml @@ -2,7 +2,7 @@ apiVersion: workload.codeflare.dev/v1beta2 kind: AppWrapper metadata: name: sample-pod - annotations: + labels: kueue.x-k8s.io/queue-name: user-queue spec: components: diff --git a/samples/wrapped-pytorch-job.yaml b/samples/wrapped-pytorch-job.yaml index a2d4dc2..5577325 100644 --- a/samples/wrapped-pytorch-job.yaml +++ b/samples/wrapped-pytorch-job.yaml @@ -2,7 +2,7 @@ apiVersion: workload.codeflare.dev/v1beta2 kind: AppWrapper metadata: name: sample-pytorch-job - annotations: + labels: kueue.x-k8s.io/queue-name: user-queue spec: components: diff --git a/site/_pages/sample-batch-job.md b/site/_pages/sample-batch-job.md index 62236d6..5a91cf8 100644 --- a/site/_pages/sample-batch-job.md +++ b/site/_pages/sample-batch-job.md @@ -10,7 +10,7 @@ apiVersion: workload.codeflare.dev/v1beta2 kind: AppWrapper metadata: name: sample-job - annotations: + labels: kueue.x-k8s.io/queue-name: user-queue spec: components: diff --git a/site/_pages/sample-pytorch.md b/site/_pages/sample-pytorch.md index 25a332c..63362e1 100644 --- a/site/_pages/sample-pytorch.md +++ b/site/_pages/sample-pytorch.md @@ -10,7 +10,7 @@ apiVersion: workload.codeflare.dev/v1beta2 kind: AppWrapper metadata: name: sample-pytorch-job - annotations: + labels: kueue.x-k8s.io/queue-name: user-queue spec: components: diff --git a/test/e2e/appwrapper_test.go b/test/e2e/appwrapper_test.go index 1a94237..536363d 100644 --- a/test/e2e/appwrapper_test.go +++ b/test/e2e/appwrapper_test.go @@ -266,6 +266,9 @@ var _ = Describe("AppWrapper E2E Test", func() { It("A failed Batch Job yields a failed AppWrapper", func() { aw := toAppWrapper(failingBatchjob(500)) + if aw.Annotations == nil { + aw.Annotations = make(map[string]string) + } aw.Annotations[workloadv1beta2.FailureGracePeriodDurationAnnotation] = "0s" aw.Annotations[workloadv1beta2.RetryLimitAnnotation] = "0" Expect(getClient(ctx).Create(ctx, aw)).To(Succeed()) diff --git a/test/e2e/util_test.go b/test/e2e/util_test.go index 3b430a9..befa809 100644 --- a/test/e2e/util_test.go +++ b/test/e2e/util_test.go @@ -178,9 +178,9 @@ func toAppWrapper(components ...workloadv1beta2.AppWrapperComponent) *workloadv1 return &workloadv1beta2.AppWrapper{ TypeMeta: metav1.TypeMeta{APIVersion: workloadv1beta2.GroupVersion.String(), Kind: "AppWrapper"}, ObjectMeta: metav1.ObjectMeta{ - Name: randName("aw"), - Namespace: testNamespace, - Annotations: map[string]string{kc.QueueLabel: testQueueName}, + Name: randName("aw"), + Namespace: testNamespace, + Labels: map[string]string{kc.QueueLabel: testQueueName}, }, Spec: workloadv1beta2.AppWrapperSpec{Components: components}, }