Merge branch 'dataframe_sort_index_optimizations' of https://github.c…

…om/galipremsagar/cudf into dataframe_sort_index_optimizations
rapidsai · Sep 16, 2021 · 13e3e14 · 13e3e14
2 parents e0e375f + 68f1e97
commit 13e3e14
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/cudf/cudf/core/frame.py b/python/cudf/cudf/core/frame.py
@@ -494,7 +494,7 @@ def _sort_index(
                         ]
                     else:
                         labels = [self.index._get_level_label(level)]
-                    midx_data = self.index.to_frame(index=False)[labels]
+                    midx_data = cudf.DataFrame._from_data(self.index._data.select_by_label(labels))
                 inds = midx_data.argsort(
                     ascending=ascending, na_position=na_position
                 )