From 93657128a3fbbf0aa9d53434d4ca4b37722a13fb Mon Sep 17 00:00:00 2001
From: Jiaxin Shan <seedjeffwan@gmail.com>
Date: Thu, 7 Feb 2019 22:09:40 -0800
Subject: [PATCH] Use command array in mpi job

---
 kubeflow/mpi-job/prototypes/mpi-job-custom.jsonnet | 2 +-
 kubeflow/mpi-job/prototypes/mpi-job-simple.jsonnet | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/kubeflow/mpi-job/prototypes/mpi-job-custom.jsonnet b/kubeflow/mpi-job/prototypes/mpi-job-custom.jsonnet
index 5b2932ddfb5..be7738a4500 100644
--- a/kubeflow/mpi-job/prototypes/mpi-job-custom.jsonnet
+++ b/kubeflow/mpi-job/prototypes/mpi-job-custom.jsonnet
@@ -22,7 +22,7 @@ local args = params.args;
 local containerCommand =
   if command != "null" then
     {
-      command: command,
+      command: std.split(command, ","),
     }
   else {};
 local containerArgs =
diff --git a/kubeflow/mpi-job/prototypes/mpi-job-simple.jsonnet b/kubeflow/mpi-job/prototypes/mpi-job-simple.jsonnet
index b16990c7dbc..a1fd42cc87f 100644
--- a/kubeflow/mpi-job/prototypes/mpi-job-simple.jsonnet
+++ b/kubeflow/mpi-job/prototypes/mpi-job-simple.jsonnet
@@ -20,7 +20,7 @@ local args = params.args;
 local containerCommand =
   if command != "null" then
     {
-      command: command,
+      command: std.split(command, ","),
     }
   else {};
 local containerArgs =