FIX-modin-project#2322: fix getting lengths and widths of partitions

Signed-off-by: Anatoly Myachev <anatoly.myachev@intel.com>

modin/engines/base/frame/data.py

@@ -1681,6 +1681,7 @@ def _copartition(self, axis, other, how, sort, force_repartition=False):
                 and axis == 0
             ):
                 is_aligning_applied = True
+                # we shouldn't modificate modin frame' partitions directly
                 self._partitions = self._frame_mgr_cls.map_axis_partitions(
                     axis, self._partitions, lambda df: df
                 )
@@ -1721,6 +1722,14 @@ def _copartition(self, axis, other, how, sort, force_repartition=False):
             reindexed_self = self._partitions
         reindexed_other_list = []
 
+        def get_column_widths(partitions):
+            if len(partitions) > 0:
+                return [obj.width() for obj in partitions[0]]
+
+        def get_row_lengths(partitions):
+            if len(partitions.T) > 0:
+                return [obj.length() for obj in partitions.T[0]]
+
         for i in range(len(other)):
             if (
                 is_aligning_applied
@@ -1734,8 +1743,9 @@ def _copartition(self, axis, other, how, sort, force_repartition=False):
                     axis,
                     other[i]._partitions,
                     lambda df: df.reindex(joined_index, axis=axis),
-                    lengths=self._row_lengths if axis == 0 else self._column_widths,
-                    manual_partition=True,
+                    lengths=get_row_lengths(reindexed_self)
+                    if axis == 0
+                    else get_column_widths(reindexed_self),
                 )
             reindexed_other_list.append(reindexed_other)
         return reindexed_self, reindexed_other_list, joined_index

modin/engines/base/frame/partition_manager.py

@@ -215,7 +215,6 @@ def broadcast_axis_partitions(
         right,
         keep_partitioning=False,
         lengths=None,
-        manual_partition=None,
     ):
         """
         Broadcast the right partitions to left and apply a function along full axis.
@@ -228,9 +227,8 @@ def broadcast_axis_partitions(
         right : The right partitions.
         keep_partitioning : boolean. Default is False
             The flag to keep partitions for Modin Frame.
-        lengths : bool
+        lengths : list(int)
             The list of lengths to shuffle the object.
-        manual_partition : list(int)
 
         Returns
         -------
@@ -250,14 +248,23 @@ def broadcast_axis_partitions(
         # may want to line to partitioning up with another BlockPartitions object. Since
         # we don't need to maintain the partitioning, this gives us the opportunity to
         # load-balance the data as well.
+        kw = {
+            "num_splits": num_splits,
+            "other_axis_partition": right_partitions,
+        }
+        if lengths:
+            kw.update(
+                {
+                    "_lengths": lengths,
+                    "manual_partition": True,
+                }
+            )
+
         result_blocks = np.array(
             [
                 part.apply(
                     preprocessed_map_func,
-                    num_splits=num_splits,
-                    other_axis_partition=right_partitions,
-                    _lengths=lengths,
-                    manual_partition=manual_partition,
+                    **kw,
                 )
                 for part in left_partitions
             ]
@@ -303,7 +310,6 @@ def map_axis_partitions(
         map_func,
         keep_partitioning=False,
         lengths=None,
-        manual_partition=None,
     ):
         """
         Applies `map_func` to every partition.
@@ -320,7 +326,6 @@ def map_axis_partitions(
             The flag to keep partitions for Modin Frame.
         lengths : list(int)
             The list of lengths to shuffle the object.
-        manual_partition : bool
 
         Returns
         -------
@@ -339,7 +344,6 @@ def map_axis_partitions(
             keep_partitioning=keep_partitioning,
             right=None,
             lengths=lengths,
-            manual_partition=manual_partition,
         )
 
     @classmethod