diff --git a/pkg/controller.v1/paddlepaddle/envvar.go b/pkg/controller.v1/paddlepaddle/envvar.go index ff5c29a592..2bcd379642 100644 --- a/pkg/controller.v1/paddlepaddle/envvar.go +++ b/pkg/controller.v1/paddlepaddle/envvar.go @@ -59,7 +59,7 @@ func setPodEnv(obj interface{}, podTemplateSpec *corev1.PodTemplateSpec, rtype, // Ref https://stackoverflow.com/questions/59812009/what-is-the-use-of-pythonunbuffered-in-docker-file. podTemplateSpec.Spec.Containers[i].Env = append(podTemplateSpec.Spec.Containers[i].Env, corev1.EnvVar{ Name: "PYTHONUNBUFFERED", - Value: "0", + Value: "1", }) podTemplateSpec.Spec.Containers[i].Env = append(podTemplateSpec.Spec.Containers[i].Env, corev1.EnvVar{ diff --git a/pkg/controller.v1/pytorch/envvar.go b/pkg/controller.v1/pytorch/envvar.go index d3ff5880a7..808aa14f50 100644 --- a/pkg/controller.v1/pytorch/envvar.go +++ b/pkg/controller.v1/pytorch/envvar.go @@ -45,7 +45,7 @@ func setPodEnv(obj interface{}, podTemplateSpec *corev1.PodTemplateSpec, rtype, podTemplateSpec.Spec.Containers[i].Env = append( podTemplateSpec.Spec.Containers[i].Env, corev1.EnvVar{ Name: "PYTHONUNBUFFERED", - Value: "0", + Value: "1", }) // If the master is not null, then we need to set the MASTER_ADDR and RANK. diff --git a/pkg/controller.v1/xgboost/xgboost.go b/pkg/controller.v1/xgboost/xgboost.go index b8f9927aa7..1ac3378080 100644 --- a/pkg/controller.v1/xgboost/xgboost.go +++ b/pkg/controller.v1/xgboost/xgboost.go @@ -92,7 +92,7 @@ func SetPodEnv(job interface{}, podTemplate *corev1.PodTemplateSpec, rtype, inde }) podTemplate.Spec.Containers[i].Env = append(podTemplate.Spec.Containers[i].Env, corev1.EnvVar{ Name: "PYTHONUNBUFFERED", - Value: "0", + Value: "1", }) // This variables are used if it is a LightGBM job if totalReplicas > 1 {