improve readability

Signed-off-by: win5923 <[email protected]>

Author: win5923

ray-operator/controllers/ray/raycluster_controller.go

@@ -1159,20 +1159,22 @@ func (r *RayClusterReconciler) shouldRecreatePodsForUpgrade(ctx context.Context,
 		actualHash := pod.Annotations[utils.PodTemplateHashKey]
 
 		var expectedHash string
-		if nodeType == string(rayv1.HeadNode) {
+		switch rayv1.RayNodeType(nodeType) {
+		case rayv1.HeadNode:
 			expectedHash = headHash
-		} else if nodeType == string(rayv1.WorkerNode) {
+		case rayv1.WorkerNode:
 			groupName := pod.Labels[utils.RayNodeGroupLabelKey]
 			var ok bool
 			expectedHash, ok = workerHashMap[groupName]
 			if !ok {
 				logger.Info("Worker group not found in spec, skipping pod", "pod", pod.Name, "groupName", groupName)
 				continue
 			}
-		} else {
+		default:
 			continue
 		}
 
+		// Check if pod needs recreation
 		if actualHash != expectedHash {
 			logger.Info("Pod template has changed, will recreate all pods",
 				"pod", pod.Name,