fix keep_partitioning

modin-project · Jul 18, 2024 · 13bcd19 · 13bcd19
1 parent 7e9073d
commit 13bcd19
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/modin/core/dataframe/pandas/partitioning/partition_manager.py b/modin/core/dataframe/pandas/partitioning/partition_manager.py
@@ -503,7 +503,7 @@ def broadcast_axis_partitions(
         keep_partitioning=False,
         num_splits=None,
         apply_indices=None,
-        send_all_right=True,
+        broadcast_all=True,
         enumerate_partitions=False,
         lengths=None,
         apply_func_args=None,
@@ -532,7 +532,7 @@ def broadcast_axis_partitions(
             then the number of splits is preserved.
         apply_indices : list of ints, default: None
             Indices of `axis ^ 1` to apply function over.
-        send_all_right : bool, default: True
+        broadcast_all : bool, default: True
             Whether or not to pass all right axis partitions to each of the left axis partitions.
         enumerate_partitions : bool, default: False
             Whether or not to pass partition index into `apply_func`.
@@ -595,7 +595,7 @@ def broadcast_axis_partitions(
                     preprocessed_map_func,
                     *(apply_func_args if apply_func_args else []),
                     other_axis_partition=(
-                        right_partitions if send_all_right else right_partitions[i]
+                        right_partitions if broadcast_all else right_partitions[i]
                     ),
                     **kw,
                     **({"partition_idx": idx} if enumerate_partitions else {}),
@@ -698,7 +698,8 @@ def broadcast_apply(
                 left=left,
                 right=right,
                 apply_func=apply_func,
-                send_all_right=False,
+                broadcast_all=False,
+                keep_partitioning=True,
             )
         return new_partitions