fix

Signed-off-by: Anatoly Myachev <[email protected]>
anmyachev · May 12, 2024 · eb65291 · eb65291
1 parent d60e0cf
commit eb65291
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/modin/core/storage_formats/pandas/merge.py b/modin/core/storage_formats/pandas/merge.py
@@ -221,7 +221,9 @@ def map_func(left, right):  # pragma: no cover
                         else new_left.sort_rows_by_column_values(on)
                     )
 
-            return new_left if keep_index else new_left.reset_index(drop=True)
+            return (
+                new_left if keep_index or not sort else new_left.reset_index(drop=True)
+            )
         else:
             return left.default_to_pandas(pandas.DataFrame.merge, right, **kwargs)
 

diff --git a/modin/tests/pandas/utils.py b/modin/tests/pandas/utils.py
@@ -702,7 +702,7 @@ def sort_if_range_partitioning(df1, df2, comparator=None, force=False):
     if comparator is None:
         comparator = df_equals
 
-    if force and (RangePartitioning.get() or use_range_partitioning_groupby()):
+    if force or (RangePartitioning.get() or use_range_partitioning_groupby()):
         df1, df2 = sort_data(df1), sort_data(df2)
 
     comparator(df1, df2)