fix torch env typo (#1573)

kubeflow · Apr 12, 2022 · b4cea34 · b4cea34
1 parent 8c43231
commit b4cea34
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pkg/controller.v1/pytorch/elastic.go b/pkg/controller.v1/pytorch/elastic.go
@@ -50,7 +50,7 @@ const (
 	// Worker/node size related arguments.
 
 	// EnvNProcPerNode is the environment variable name for the number of processes per node.
-	EnvNProcPerNode = "PET_N_PROC_PER_NODE"
+	EnvNProcPerNode = "PET_NPROC_PER_NODE"
 	// EnvNNodes is the environment variable name for the number of nodes.
 	EnvNNodes = "PET_NNODES"
 )