Update dataframe.py

rapidsai · May 10, 2024 · 064020c · 064020c
1 parent f212752
commit 064020c
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/cudf/cudf/core/dataframe.py b/python/cudf/cudf/core/dataframe.py
@@ -357,9 +357,9 @@ def _getitem_tuple_arg(self, arg):
                     # as join is not assigning any names to index,
                     # update it over here
                     df.index.name = columns_df.index.name
-                    if not isinstance(df.index, cudf.MultiIndex) and is_numeric_dtype(
-                        df.index.dtype
-                    ):
+                    if not isinstance(
+                        df.index, MultiIndex
+                    ) and is_numeric_dtype(df.index.dtype):
                         # Preserve the original index type.
                         df.index = df.index.astype(self._frame.index.dtype)
                     df = df.sort_values(by=[tmp_col_name, cantor_name])