Remove resource limit

Signed-off-by: Yuvraj <[email protected]>
flyteorg · Oct 26, 2021 · 08c8dc4 · 08c8dc4
1 parent 1b50d41
commit 08c8dc4
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/plugins/flytekit-kf-mpi/flytekitplugins/kfmpi/task.py b/plugins/flytekit-kf-mpi/flytekitplugins/kfmpi/task.py
@@ -82,8 +82,6 @@ class MPIJob(object):
     slots: int
     num_launcher_replicas: int = 1
     num_workers: int = 1
-    per_replica_requests: Optional[_task_model.Resources] = None
-    per_replica_limits: Optional[_task_model.Resources] = None
 
 
 class MPIFunctionTask(PythonFunctionTask[MPIJob]):
@@ -119,7 +117,7 @@ def __init__(self, task_config: MPIJob, task_function: Callable, **kwargs):
             task_config=task_config,
             task_function=task_function,
             task_type=self._MPI_JOB_TASK_TYPE,
-            **{**kwargs, "requests": task_config.per_replica_requests, "limits": task_config.per_replica_limits},
+            **kwargs,
         )
 
     def get_command(self, settings: SerializationSettings) -> List[str]: