fix mypy

Signed-off-by: YunLiu <[email protected]>

Author: KumoLiu

monai/transforms/compose.py

@@ -66,7 +66,7 @@ def execute_compose(
         data: a tensor-like object to be transformed
         transforms: a sequence of transforms to be carried out
         map_items: whether to apply transform to each item in the input `data` if `data` is a list or tuple.
-            defaults to `True`. If set to an integer, the transform will be applied to that index of the input `data`.
+            defaults to `True`. If set to an integer, recursively map the items in `data` `map_items` times.
         unpack_items: whether to unpack input `data` with `*` as parameters for the callable function of transform.
             defaults to `False`.
         start: the index of the first transform to be executed. If not set, this defaults to 0
@@ -206,7 +206,7 @@ class Compose(Randomizable, InvertibleTransform, LazyTransform):
     Args:
         transforms: sequence of callables.
         map_items: whether to apply transform to each item in the input `data` if `data` is a list or tuple.
-            defaults to `True`. If set to an integer, the transform will be applied to that index of the input `data`.
+            defaults to `True`. If set to an integer, recursively map the items in `data` `map_items` times.
         unpack_items: whether to unpack input `data` with `*` as parameters for the callable function of transform.
             defaults to `False`.
         log_stats: this optional parameter allows you to specify a logger by name for logging of pipeline execution.
@@ -392,7 +392,7 @@ class OneOf(Compose):
         weights: probabilities corresponding to each callable in transforms.
             Probabilities are normalized to sum to one.
         map_items: whether to apply transform to each item in the input `data` if `data` is a list or tuple.
-            defaults to `True`.
+            defaults to `True`. If set to an integer, recursively map the items in `data` `map_items` times.
         unpack_items: whether to unpack input `data` with `*` as parameters for the callable function of transform.
             defaults to `False`.
         log_stats: this optional parameter allows you to specify a logger by name for logging of pipeline execution.
@@ -414,7 +414,7 @@ def __init__(
         self,
         transforms: Sequence[Callable] | Callable | None = None,
         weights: Sequence[float] | float | None = None,
-        map_items: bool = True,
+        map_items: bool | int = True,
         unpack_items: bool = False,
         log_stats: bool | str = False,
         lazy: bool | None = False,

monai/transforms/transform.py

@@ -106,7 +106,7 @@ def apply_transform(
     log_stats: bool | str = False,
     lazy: bool | None = None,
     overrides: dict | None = None,
-) -> list[ReturnType] | ReturnType:
+) -> list[Any] | ReturnType:
     """
     Transform `data` with `transform`.
 
@@ -119,7 +119,7 @@ def apply_transform(
         data: an object to be transformed.
         map_items: whether to apply transform to each item in `data`,
             if `data` is a list or tuple. Defaults to True.
-            it can also be an int, if so, apply the transform to each item in the list `map_items` times.
+            it can also be an int, if so, recursively map the items in `data` `map_items` times.
         unpack_items: whether to unpack parameters using `*`. Defaults to False.
         log_stats: log errors when they occur in the processing pipeline. By default, this is set to False, which
             disables the logger for processing pipeline errors. Setting it to None or True will enable logging to the