Skip to content

Add PodTemplateOverrides API into TrainJob #2784

Description

@mimowo

What you would like to be added?

I would like to add Labels and Annotations to the .spec.podSpecOverrides in TrainJob, which will be propagated down to Pods per Job.

Why is this needed?

In order to support TrainerJob fully in Kueue:

The label & annotation are set by Kueue per-Job.

Maybe we could rewrite the integration in Kueue without the label and the annotation, but I'm not sure how, and it requires extra effort.

Since all other Job CRDs Kueue integrates with (Job, JobSet, KubeRay, old Kubeflow ...) allow setting Labels and Annotations per PodTemplate I would also expect it from the new Kubeflow Trainer.

Love this feature?

Give it a 👍 We prioritize the features with most 👍

Metadata

Metadata

Assignees

Type

No type

Fields

No fields configured for issues without a type.

Projects

No projects

Relationships

None yet

Development

No branches or pull requests

Issue actions