chore(plugin): Update appending args logic in jobset builder.

Signed-off-by: Electronic-Waste <[email protected]>

Author: Electronic-Waste

pkg/runtime/framework/plugins/jobset/builder.go

@@ -127,9 +127,6 @@ func (b *Builder) Trainer(info *runtime.Info, trainJob *trainer.TrainJob) *Build
 						if command := jobTrainer.Command; command != nil {
 							b.Spec.ReplicatedJobs[i].Template.Spec.Template.Spec.Containers[j].Command = command
 						}
-						if args := jobTrainer.Args; args != nil {
-							b.Spec.ReplicatedJobs[i].Template.Spec.Template.Spec.Containers[j].Args = args
-						}
 						if resourcesPerNode := jobTrainer.ResourcesPerNode; resourcesPerNode != nil &&
 							(resourcesPerNode.Limits != nil || resourcesPerNode.Requests != nil) {
 							requirements := corev1ac.ResourceRequirements()
@@ -142,6 +139,10 @@ func (b *Builder) Trainer(info *runtime.Info, trainJob *trainer.TrainJob) *Build
 							b.Spec.ReplicatedJobs[i].Template.Spec.Template.Spec.Containers[j].
 								WithResources(requirements)
 						}
+						apply.UpsertArgs(
+							&b.Spec.ReplicatedJobs[i].Template.Spec.Template.Spec.Containers[j].Args,
+							jobTrainer.Args...,
+						)
 						apply.UpsertEnvVars(
 							&b.Spec.ReplicatedJobs[i].Template.Spec.Template.Spec.Containers[j].Env,
 							apply.EnvVars(jobTrainer.Env...)...,