cmd changes

Signed-off-by: Yuvraj <[email protected]>
flyteorg · Oct 22, 2021 · bd60e40 · bd60e40
1 parent 2e43df6
commit bd60e40
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/plugins/flytekit-kf-mpi/flytekitplugins/kfmpi/task.py b/plugins/flytekit-kf-mpi/flytekitplugins/kfmpi/task.py
@@ -129,7 +129,7 @@ def get_command(self, settings: SerializationSettings) -> List[str]:
             self._MPI_BASE_COMMAND
             + ["-np", f"{num_procs}"]
             + ["python", settings.entrypoint_settings.path]
-            + cmd[1:]
+            + cmd
         )
         # the hostfile is set automatically by MPIOperator using env variable OMPI_MCA_orte_default_hostfile
         return mpi_cmd