update label for tf-controller (#1474)

kubeflow · Nov 21, 2021 · 9894d67 · 9894d67
1 parent 633ec1f
commit 9894d67
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pkg/controller.v1/tensorflow/tfjob_controller.go b/pkg/controller.v1/tensorflow/tfjob_controller.go
@@ -794,6 +794,8 @@ func (r *TFJobReconciler) createNewPod(tfjob *tfv1.TFJob, rt, index string, spec
 	labels := r.GenLabels(tfjob.Name)
 	labels[tfReplicaTypeLabel] = rt
 	labels[tfReplicaIndexLabel] = index
+	labels[commonv1.ReplicaTypeLabel] = rt
+	labels[commonv1.ReplicaIndexLabel] = index
 
 	if masterRole {
 		labels[commonv1.JobRoleLabel] = "master"