@@ -520,18 +520,18 @@ func (j *JobSetWrapper) PodPriorityClassName(value string) *JobSetWrapper {
520520}
521521
522522func (j * JobSetWrapper ) Label (key , value string ) * JobSetWrapper {
523- if j .ObjectMeta . Labels == nil {
524- j .ObjectMeta . Labels = make (map [string ]string , 1 )
523+ if j .Labels == nil {
524+ j .Labels = make (map [string ]string , 1 )
525525 }
526- j .ObjectMeta . Labels [key ] = value
526+ j .Labels [key ] = value
527527 return j
528528}
529529
530530func (j * JobSetWrapper ) Annotation (key , value string ) * JobSetWrapper {
531- if j .ObjectMeta . Annotations == nil {
532- j .ObjectMeta . Annotations = make (map [string ]string , 1 )
531+ if j .Annotations == nil {
532+ j .Annotations = make (map [string ]string , 1 )
533533 }
534- j .ObjectMeta . Annotations [key ] = value
534+ j .Annotations [key ] = value
535535 return j
536536}
537537
@@ -665,10 +665,10 @@ func (t *TrainJobTrainerWrapper) NumProcPerNode(numProcPerNode intstr.IntOrStrin
665665}
666666
667667func (t * TrainJobTrainerWrapper ) Container (image string , command []string , args []string , resRequests corev1.ResourceList ) * TrainJobTrainerWrapper {
668- t .Trainer . Image = & image
669- t .Trainer . Command = command
670- t .Trainer . Args = args
671- t .Trainer . ResourcesPerNode = & corev1.ResourceRequirements {
668+ t .Image = & image
669+ t .Command = command
670+ t .Args = args
671+ t .ResourcesPerNode = & corev1.ResourceRequirements {
672672 Requests : resRequests ,
673673 }
674674 return t
@@ -694,12 +694,12 @@ func MakeTrainJobInitializerWrapper() *TrainJobInitializerWrapper {
694694}
695695
696696func (t * TrainJobInitializerWrapper ) DatasetInitializer (datasetInitializer * trainer.DatasetInitializer ) * TrainJobInitializerWrapper {
697- t .Initializer . Dataset = datasetInitializer
697+ t .Dataset = datasetInitializer
698698 return t
699699}
700700
701701func (t * TrainJobInitializerWrapper ) ModelInitializer (modelInitializer * trainer.ModelInitializer ) * TrainJobInitializerWrapper {
702- t .Initializer . Model = modelInitializer
702+ t .Model = modelInitializer
703703 return t
704704}
705705
@@ -901,18 +901,18 @@ func MakeTrainingRuntimeWrapper(namespace, name string) *TrainingRuntimeWrapper
901901}
902902
903903func (r * TrainingRuntimeWrapper ) Label (key , value string ) * TrainingRuntimeWrapper {
904- if r .ObjectMeta . Labels == nil {
905- r .ObjectMeta . Labels = make (map [string ]string , 1 )
904+ if r .Labels == nil {
905+ r .Labels = make (map [string ]string , 1 )
906906 }
907- r .ObjectMeta . Labels [key ] = value
907+ r .Labels [key ] = value
908908 return r
909909}
910910
911911func (r * TrainingRuntimeWrapper ) Annotation (key , value string ) * TrainingRuntimeWrapper {
912- if r .ObjectMeta . Annotations == nil {
913- r .ObjectMeta . Annotations = make (map [string ]string , 1 )
912+ if r .Annotations == nil {
913+ r .Annotations = make (map [string ]string , 1 )
914914 }
915- r .ObjectMeta . Annotations [key ] = value
915+ r .Annotations [key ] = value
916916 return r
917917}
918918
@@ -1289,17 +1289,17 @@ func MakeSchedulerPluginsPodGroup(namespace, name string) *SchedulerPluginsPodGr
12891289}
12901290
12911291func (p * SchedulerPluginsPodGroupWrapper ) MinMember (members int32 ) * SchedulerPluginsPodGroupWrapper {
1292- p .PodGroup . Spec .MinMember = members
1292+ p .Spec .MinMember = members
12931293 return p
12941294}
12951295
12961296func (p * SchedulerPluginsPodGroupWrapper ) MinResources (resources corev1.ResourceList ) * SchedulerPluginsPodGroupWrapper {
1297- p .PodGroup . Spec .MinResources = resources
1297+ p .Spec .MinResources = resources
12981298 return p
12991299}
13001300
13011301func (p * SchedulerPluginsPodGroupWrapper ) SchedulingTimeout (timeout int32 ) * SchedulerPluginsPodGroupWrapper {
1302- p .PodGroup . Spec .ScheduleTimeoutSeconds = & timeout
1302+ p .Spec .ScheduleTimeoutSeconds = & timeout
13031303 return p
13041304}
13051305
@@ -1339,27 +1339,27 @@ func MakeVolcanoPodGroup(namespace, name string) *VolcanoPodGroupWrapper {
13391339}
13401340
13411341func (p * VolcanoPodGroupWrapper ) MinMember (members int32 ) * VolcanoPodGroupWrapper {
1342- p .PodGroup . Spec .MinMember = members
1342+ p .Spec .MinMember = members
13431343 return p
13441344}
13451345
13461346func (p * VolcanoPodGroupWrapper ) MinResources (resources * corev1.ResourceList ) * VolcanoPodGroupWrapper {
1347- p .PodGroup . Spec .MinResources = resources
1347+ p .Spec .MinResources = resources
13481348 return p
13491349}
13501350
13511351func (p * VolcanoPodGroupWrapper ) Queue (queue string ) * VolcanoPodGroupWrapper {
1352- p .PodGroup . Spec .Queue = queue
1352+ p .Spec .Queue = queue
13531353 return p
13541354}
13551355
13561356func (p * VolcanoPodGroupWrapper ) PriorityClassName (pc string ) * VolcanoPodGroupWrapper {
1357- p .PodGroup . Spec .PriorityClassName = pc
1357+ p .Spec .PriorityClassName = pc
13581358 return p
13591359}
13601360
13611361func (p * VolcanoPodGroupWrapper ) NetworkTopology (mode volcanov1beta1.NetworkTopologyMode , highestTier int ) * VolcanoPodGroupWrapper {
1362- p .PodGroup . Spec .NetworkTopology = & volcanov1beta1.NetworkTopologySpec {
1362+ p .Spec .NetworkTopology = & volcanov1beta1.NetworkTopologySpec {
13631363 Mode : mode ,
13641364 HighestTierAllowed : & highestTier ,
13651365 }
@@ -1374,7 +1374,7 @@ func (p *VolcanoPodGroupWrapper) ControllerReference(gvk schema.GroupVersionKind
13741374 UID : types .UID (uid ),
13751375 },
13761376 }, gvk )
1377- p .PodGroup . OwnerReferences = append (p . PodGroup .OwnerReferences , owner )
1377+ p .OwnerReferences = append (p .OwnerReferences , owner )
13781378 return p
13791379}
13801380
0 commit comments