Skip to content

Commit

Permalink
revert undesired changes
Browse files Browse the repository at this point in the history
Signed-off-by: Dmitry Chigarev <dmitry.chigarev@intel.com>
  • Loading branch information
dchigarev committed Jan 29, 2024
1 parent 7bda8ea commit 1aba0fa
Showing 1 changed file with 7 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -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*.
Expand Down Expand Up @@ -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
Expand Down

0 comments on commit 1aba0fa

Please sign in to comment.