diff --git a/modin/core/dataframe/pandas/partitioning/partition_manager.py b/modin/core/dataframe/pandas/partitioning/partition_manager.py index e9494aeec10..f7b3899550c 100644 --- a/modin/core/dataframe/pandas/partitioning/partition_manager.py +++ b/modin/core/dataframe/pandas/partitioning/partition_manager.py @@ -612,7 +612,12 @@ def map_partitions( @classmethod @wait_computations_if_benchmark_mode def lazy_map_partitions( - cls, partitions, map_func, func_args=None, enumerate_partitions=False + cls, + partitions, + map_func, + func_args=None, + func_kwargs=None, + enumerate_partitions=False, ): """ Apply `map_func` to every partition in `partitions` *lazily*. @@ -641,6 +646,7 @@ def lazy_map_partitions( part.add_to_apply_calls( preprocessed_map_func, *(tuple() if func_args is None else func_args), + **func_kwargs if func_kwargs is not None else {}, **({"partition_idx": i} if enumerate_partitions else {}), ) for part in row