Address PR comments.

rapidsai · Feb 23, 2022 · ae168f4 · ae168f4
1 parent 102ccef
commit ae168f4
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/python/cudf/cudf/core/groupby/groupby.py b/python/cudf/cudf/core/groupby/groupby.py
@@ -1615,9 +1615,8 @@ def agg(self, func):
                 return result.iloc[:, 0]
 
         # drop the first level if we have a multiindex
-        cols = result._data.to_pandas_index()
-        if result._data.multiindex and cols.nlevels > 1:
-            result.columns = cols.droplevel(0)
+        if result._data.nlevels > 1:
+            result.columns = result._data.to_pandas_index().droplevel(0)
 
         return result