Update modin/core/dataframe/pandas/dataframe/dataframe.py

Co-authored-by: Dmitry Chigarev <[email protected]>
YarShev · Mar 28, 2024 · 478f137 · 478f137
1 parent c7bfb24
commit 478f137
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modin/core/dataframe/pandas/dataframe/dataframe.py b/modin/core/dataframe/pandas/dataframe/dataframe.py
@@ -3450,7 +3450,7 @@ def broadcast_apply_full_axis(
                 kw["column_widths"] = [len(new_columns)]
         elif axis == 1:
             if is_index_materialized and new_partitions.shape[0] == 1:
-                kw["row_lengths"] = [len(new_columns)]
+                kw["row_lengths"] = [len(new_index)]
         if not keep_partitioning:
             if kw["row_lengths"] is None and is_index_materialized:
                 if axis == 0: