Skip to content

Commit 8d354a7

Browse files
committed
fix: copy spec.Containers[0].Env
1 parent 3b88c53 commit 8d354a7

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

internal/worker/worker.go

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,12 +25,13 @@ func (wg *WorkerGenerator) GenerateWorkerPod(
2525
) *corev1.Pod {
2626

2727
spec := wg.WorkerConfig.Template.Spec
28+
2829
if spec.NodeSelector == nil {
2930
spec.NodeSelector = make(map[string]string)
3031
}
3132
spec.NodeSelector = gpu.Status.NodeSelector
3233

33-
spec.Containers[0].Env = append(spec.Containers[0].Env, corev1.EnvVar{
34+
spec.Containers[0].Env = append(append([]corev1.EnvVar{}, spec.Containers[0].Env...), corev1.EnvVar{
3435
Name: "NVIDIA_VISIBLE_DEVICES",
3536
Value: gpu.Status.UUID,
3637
})

0 commit comments

Comments
 (0)