[feature] Does tfJob support setting different label/envVar for each worker(replicas >1)?

[sdf611097]

In my case, I have some tools which are designed by pod label. How can I set different label for each pod which is created by tfJob? (N workers with different custom labels)

    - replicas: 2
      tfReplicaType: WORKER
      template:
        spec:
          containers:
            - image: gcr.io/tf-on-k8s-dogfood/tf_sample:dc944ff
              name: tensorflow
          restartPolicy: OnFailure
          #replicaLabels: 
          #- labelA: valueA-0
          #  labelB: valueB-0
          #- labelA: valueA-1
          #  labelB: valueB-1

If not, are there any hints or suggestions to do this(Maybe I can make a PR). Such as naming, how to validating(spec.replicaSpecs.replicas equal to replicaLabels size) and so on.

[gaocegege]

@sdf611097

You could add labels in the YAML file and I think it should be added into the jobs and services created by the TFJob. If not I think it is a bug.

[sdf611097]

@gaocegege

I can set same labels to N workers as expected, but I don't know how to set different label for each worker. For example, I have 2 worker, each worker need 2 label (labelA, labelB).

worker0:
   labelA: valueA-0
   labelB: valueB-0
worker1:
   labelA: valueA-1
   labelB: valueB-1

P.S. I think my question is not clearly, so I made some modifications.

[gaocegege]

@sdf611097

Got it, then it is a new feature.

[jlewi]

You can't add individual labels specific to a replica (e.g. worker 0 vs. worker1). We do however add standard labels corresponding to the replica type and replica index.

[jlewi]

@sdf611097 Does this address your needs?

[sdf611097]

Yes. If I still need to set different label or env var for each job, I will send a PR.

[jlewi]

@sdf611097 Thanks.