From 13bcd19fd8a8b232ab64df0b821c33703a15fb51 Mon Sep 17 00:00:00 2001 From: Kirill Suvorov Date: Thu, 18 Jul 2024 09:23:14 +0000 Subject: [PATCH] fix keep_partitioning --- .../dataframe/pandas/partitioning/partition_manager.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/modin/core/dataframe/pandas/partitioning/partition_manager.py b/modin/core/dataframe/pandas/partitioning/partition_manager.py index 6553074a7b9..bd3a1d14760 100644 --- 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