Skip to content

Commit

Permalink
Fix tests
Browse files Browse the repository at this point in the history
  • Loading branch information
Retribution98 committed Apr 16, 2024
1 parent c5d2ca7 commit 5aa374c
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 3 deletions.
2 changes: 1 addition & 1 deletion modin/core/dataframe/pandas/dataframe/dataframe.py
Original file line number Diff line number Diff line change
Expand Up @@ -2232,7 +2232,7 @@ def map(
func,
keep_partitioning=True,
map_func_args=func_args,
**func_kwargs if func_kwargs is not None else {},
map_func_kwargs=func_kwargs,
)
else:
new_partitions = (

Check warning on line 2238 in modin/core/dataframe/pandas/dataframe/dataframe.py

View check run for this annotation

Codecov / codecov/patch

modin/core/dataframe/pandas/dataframe/dataframe.py#L2238

Added line #L2238 was not covered by tests
Expand Down
12 changes: 10 additions & 2 deletions modin/core/dataframe/pandas/partitioning/partition_manager.py
Original file line number Diff line number Diff line change
Expand Up @@ -494,6 +494,7 @@ def broadcast_axis_partitions(
enumerate_partitions=False,
lengths=None,
apply_func_args=None,
apply_func_kwargs=None,
**kwargs,
):
"""
Expand Down Expand Up @@ -529,6 +530,8 @@ def broadcast_axis_partitions(
2. When passing lengths you must explicitly specify `keep_partitioning=False`.
apply_func_args : list-like, optional
Positional arguments to pass to the `func`.
apply_func_kwargs : dict, optional
Keyword arguments to pass to the `func`.
**kwargs : dict
Additional options that could be used by different engines.
Expand Down Expand Up @@ -580,6 +583,7 @@ def broadcast_axis_partitions(
left_partitions[i].apply(
preprocessed_map_func,
*(apply_func_args if apply_func_args else []),
**apply_func_kwargs if apply_func_kwargs is not None else {},
**kw,
**({"partition_idx": idx} if enumerate_partitions else {}),
**kwargs,
Expand Down Expand Up @@ -692,6 +696,7 @@ def map_axis_partitions(
lengths=None,
enumerate_partitions=False,
map_func_args=None,
map_func_kwargs=None,
**kwargs,
):
"""
Expand Down Expand Up @@ -723,6 +728,8 @@ def map_axis_partitions(
Note that `map_func` must be able to accept `partition_idx` kwarg.
map_func_args : list-like, optional
Positional arguments to pass to the `map_func`.
map_func_kwargs : dict, optional
Keyword arguments for the 'map_func'.
**kwargs : dict
Additional options that could be used by different engines.
Expand All @@ -746,6 +753,7 @@ def map_axis_partitions(
lengths=lengths,
enumerate_partitions=enumerate_partitions,
apply_func_args=map_func_args,
apply_func_kwargs=map_func_kwargs,
**kwargs,
)

Expand All @@ -769,9 +777,9 @@ def map_partitions_splitting_by_column(
The number of splits by column.
map_func : callable
Function to apply.
func_args : iterable, optional
map_func_args : iterable, optional
Positional arguments for the 'map_func'.
func_kwargs : dict, optional
map_func_kwargs : dict, optional
Keyword arguments for the 'map_func'.
Returns
Expand Down

0 comments on commit 5aa374c

Please sign in to comment.