make lint

Signed-off-by: Yuvraj <[email protected]>
flyteorg · Oct 22, 2021 · 1b50d41 · 1b50d41
1 parent bd60e40
commit 1b50d41
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/plugins/flytekit-kf-mpi/flytekitplugins/kfmpi/task.py b/plugins/flytekit-kf-mpi/flytekitplugins/kfmpi/task.py
@@ -125,12 +125,7 @@ def __init__(self, task_config: MPIJob, task_function: Callable, **kwargs):
     def get_command(self, settings: SerializationSettings) -> List[str]:
         cmd = super().get_command(settings)
         num_procs = self.task_config.num_workers * self.task_config.slots
-        mpi_cmd = (
-            self._MPI_BASE_COMMAND
-            + ["-np", f"{num_procs}"]
-            + ["python", settings.entrypoint_settings.path]
-            + cmd
-        )
+        mpi_cmd = self._MPI_BASE_COMMAND + ["-np", f"{num_procs}"] + ["python", settings.entrypoint_settings.path] + cmd
         # the hostfile is set automatically by MPIOperator using env variable OMPI_MCA_orte_default_hostfile
         return mpi_cmd