Updating variable names in runtime tranformation method

Signed-off-by: Vishesh Tanksale <[email protected]>
NVIDIA · Jul 11, 2024 · 3d91ca3 · 3d91ca3
1 parent 993f7cf
commit 3d91ca3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/controllers/object_controls.go b/controllers/object_controls.go
@@ -1254,16 +1254,16 @@ func TransformToolkit(obj *appsv1.DaemonSet, config *gpuv1.ClusterPolicySpec, n
 	return nil
 }
 
-func transformForRuntime(obj *appsv1.DaemonSet, config *gpuv1.ClusterPolicySpec, runtime string, mainContainerName string) error {
+func transformForRuntime(obj *appsv1.DaemonSet, config *gpuv1.ClusterPolicySpec, runtime string, containerName string) error {
 	var mainContainer *corev1.Container
 	for i, ctr := range obj.Spec.Template.Spec.Containers {
-		if ctr.Name == mainContainerName {
+		if ctr.Name == containerName {
 			mainContainer = &obj.Spec.Template.Spec.Containers[i]
 			break
 		}
 	}
 	if mainContainer == nil {
-		return fmt.Errorf("failed to find main container %q", mainContainerName)
+		return fmt.Errorf("failed to find main container %q", containerName)
 	}
 
 	setContainerEnv(mainContainer, "RUNTIME", runtime)