Skip to content

Commit 13bcd19

Browse files
committed
fix keep_partitioning
1 parent 7e9073d commit 13bcd19

File tree

1 file changed

+5
-4
lines changed

1 file changed

+5
-4
lines changed

modin/core/dataframe/pandas/partitioning/partition_manager.py

+5-4
Original file line numberDiff line numberDiff line change
@@ -503,7 +503,7 @@ def broadcast_axis_partitions(
503503
keep_partitioning=False,
504504
num_splits=None,
505505
apply_indices=None,
506-
send_all_right=True,
506+
broadcast_all=True,
507507
enumerate_partitions=False,
508508
lengths=None,
509509
apply_func_args=None,
@@ -532,7 +532,7 @@ def broadcast_axis_partitions(
532532
then the number of splits is preserved.
533533
apply_indices : list of ints, default: None
534534
Indices of `axis ^ 1` to apply function over.
535-
send_all_right : bool, default: True
535+
broadcast_all : bool, default: True
536536
Whether or not to pass all right axis partitions to each of the left axis partitions.
537537
enumerate_partitions : bool, default: False
538538
Whether or not to pass partition index into `apply_func`.
@@ -595,7 +595,7 @@ def broadcast_axis_partitions(
595595
preprocessed_map_func,
596596
*(apply_func_args if apply_func_args else []),
597597
other_axis_partition=(
598-
right_partitions if send_all_right else right_partitions[i]
598+
right_partitions if broadcast_all else right_partitions[i]
599599
),
600600
**kw,
601601
**({"partition_idx": idx} if enumerate_partitions else {}),
@@ -698,7 +698,8 @@ def broadcast_apply(
698698
left=left,
699699
right=right,
700700
apply_func=apply_func,
701-
send_all_right=False,
701+
broadcast_all=False,
702+
keep_partitioning=True,
702703
)
703704
return new_partitions
704705

0 commit comments

Comments
 (0)