diff --git a/flytekit/core/map_task.py b/flytekit/core/map_task.py index 9e62b6d96c..42645a4797 100644 --- a/flytekit/core/map_task.py +++ b/flytekit/core/map_task.py @@ -220,8 +220,8 @@ def map_task(task_function: PythonFunctionTask, concurrency: int = None, min_suc :language: python :dedent: 4 - At run time, the underlying map task will be run for every value in the input collection. Task-specific attributes - such as :py:class:`flytekit.TaskMetadata` and :py:class:`flytekit.Resources` are applied to individual instances + At run time, the underlying map task will be run for every value in the input collection. Attributes + such as :py:class:`flytekit.TaskMetadata` and ``with_overrides`` are applied to individual instances of the mapped task. :param task_function: This argument is implicitly passed and represents the repeatable function @@ -231,10 +231,6 @@ def map_task(task_function: PythonFunctionTask, concurrency: int = None, min_suc :param min_success_ratio: If specified, this determines the minimum fraction of total jobs which can complete successfully before terminating this task and marking it successful. - ``with_overrides`` on a map task can be used to set individual map task resource assigment. - - .. code-block:: python - map_task(my_mappable_task)(...).with_overrides(requests=ResourceRequests(cpu="1", memory="300Mi"), retries=1) """ if not isinstance(task_function, PythonFunctionTask): raise ValueError( diff --git a/tests/flytekit/unit/core/test_map_task.py b/tests/flytekit/unit/core/test_map_task.py index 95df669829..31cbceffe9 100644 --- a/tests/flytekit/unit/core/test_map_task.py +++ b/tests/flytekit/unit/core/test_map_task.py @@ -3,7 +3,7 @@ import pytest -from flytekit import LaunchPlan, Resources, map_task +from flytekit import LaunchPlan, map_task from flytekit.common.translator import get_serializable from flytekit.core import context_manager from flytekit.core.context_manager import Image, ImageConfig @@ -33,13 +33,9 @@ def my_mappable_task(a: int) -> str: @workflow def my_wf(x: typing.List[int]) -> typing.List[str]: - return map_task( - my_mappable_task, - metadata=TaskMetadata(retries=1), - requests=Resources(cpu="10M"), - concurrency=10, - min_success_ratio=0.75, - )(a=x) + return map_task(my_mappable_task, metadata=TaskMetadata(retries=1), concurrency=10, min_success_ratio=0.75,)( + a=x + ).with_overrides(cpu="10M") # test_map_task_end