fix

Signed-off-by: Anatoly Myachev <anatoly.myachev@intel.com>
modin-project · Apr 12, 2024 · b589d27 · b589d27
1 parent 40b5569
commit b589d27
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 5 deletions.
diff --git a/modin/core/execution/dask/implementations/pandas_on_dask/partitioning/virtual_partition.py b/modin/core/execution/dask/implementations/pandas_on_dask/partitioning/virtual_partition.py
@@ -160,9 +160,9 @@ def deploy_axis_func(
                 *partitions,
             ),
             f_kwargs={
+                "min_block_size": MinPartitionSize.get(),
                 "lengths": lengths,
                 "manual_partition": manual_partition,
-                "min_block_size": MinPartitionSize.get(),
             },
             num_returns=result_num_splits * (1 + cls._PARTITIONS_METADATA_LEN),
             pure=False,

diff --git a/modin/core/execution/ray/implementations/pandas_on_ray/partitioning/virtual_partition.py b/modin/core/execution/ray/implementations/pandas_on_ray/partitioning/virtual_partition.py
@@ -189,8 +189,8 @@ def deploy_axis_func(
             f_len_args=len(f_args),
             f_kwargs=f_kwargs,
             manual_partition=manual_partition,
-            lengths=lengths,
             min_block_size=MinPartitionSize.get(),
+            lengths=lengths,
             return_generator=True,
         )
 

diff --git a/...ore/execution/unidist/implementations/pandas_on_unidist/partitioning/virtual_partition.py b/...ore/execution/unidist/implementations/pandas_on_unidist/partitioning/virtual_partition.py
@@ -189,8 +189,8 @@ def deploy_axis_func(
             maintain_partitioning,
             *partitions,
             manual_partition=manual_partition,
-            lengths=lengths,
             min_block_size=MinPartitionSize.get(),
+            lengths=lengths,
         )
 
     @classmethod

diff --git a/modin/core/storage_formats/pandas/utils.py b/modin/core/storage_formats/pandas/utils.py
@@ -22,7 +22,7 @@
 import numpy as np
 import pandas
 
-from modin.config import NPartitions
+from modin.config import MinPartitionSize, NPartitions
 
 
 def compute_chunksize(axis_len: int, num_splits: int, min_block_size: int) -> int:
@@ -251,7 +251,11 @@ def merge_partitioning(left, right, axis=1):
 
     if lshape is not None and rshape is not None:
         res_shape = sum(lshape) + sum(rshape)
-        chunk_size = compute_chunksize(axis_len=res_shape, num_splits=NPartitions.get())
+        chunk_size = compute_chunksize(
+            axis_len=res_shape,
+            num_splits=NPartitions.get(),
+            min_block_size=MinPartitionSize.get(),
+        )
         return ceil(res_shape / chunk_size)
     else:
         lsplits = left._partitions.shape[axis]