Fix Master Label for PyTorchJob

andreyvelich · Jan 4, 2024 · 329b21f · 329b21f
1 parent ab5de23
commit 329b21f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pkg/controller.v1/pytorch/pytorchjob_controller.go b/pkg/controller.v1/pytorch/pytorchjob_controller.go
@@ -507,10 +507,10 @@ func (r *PyTorchJobReconciler) SetClusterSpec(job interface{}, podTemplate *core
 func (r *PyTorchJobReconciler) IsMasterRole(replicas map[kubeflowv1.ReplicaType]*kubeflowv1.ReplicaSpec,
 	rtype kubeflowv1.ReplicaType, index int) bool {
 	if _, ok := replicas[kubeflowv1.PyTorchJobReplicaTypeMaster]; ok {
-		return string(rtype) == strings.ToLower(string(kubeflowv1.PyTorchJobReplicaTypeMaster))
+		return rtype == kubeflowv1.PyTorchJobReplicaTypeMaster
 	}
 	// else check if it is worker with index 0
-	return string(rtype) == strings.ToLower(string(kubeflowv1.PyTorchJobReplicaTypeWorker)) && index == 0
+	return rtype == kubeflowv1.PyTorchJobReplicaTypeWorker && index == 0
 }
 
 func (r *PyTorchJobReconciler) GetDefaultContainerName() string {