From 329b21f6b440c7775720fa255b49b8172230811c Mon Sep 17 00:00:00 2001
From: Andrey Velichkevich <andrey.velichkevich@gmail.com>
Date: Thu, 4 Jan 2024 20:09:51 +0000
Subject: [PATCH] Fix Master Label for PyTorchJob

---
 pkg/controller.v1/pytorch/pytorchjob_controller.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pkg/controller.v1/pytorch/pytorchjob_controller.go b/pkg/controller.v1/pytorch/pytorchjob_controller.go
index 13cbee9889..f50821eaf1 100644
--- 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 {