fix issue

rapidsai · Feb 15, 2023 · a83c5dc · a83c5dc
1 parent 029b040
commit a83c5dc
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/python/cudf/cudf/core/groupby/groupby.py b/python/cudf/cudf/core/groupby/groupby.py
@@ -297,14 +297,15 @@ def dtypes(self):
         2  object  int64
         3  object  int64
         """
-        non_grouped = self.grouping.values._column_names
-        return pd.DataFrame(
+        index = self.grouping.keys.unique().to_pandas()
+        result = pd.DataFrame(
             {
                 name: pd.Series([self.obj._dtypes[name]]).repeat(len(index))
-                for name in non_grouped
+                for name in self.grouping.values._column_names
             },
-            index=index
         )
+        result.index = index
+        return result
 
     @cached_property
     def groups(self):