Add correct groups object

Author: benjaminarjun

pandas/core/groupby/ops.py

@@ -241,7 +241,7 @@ def groups(self):
         else:
             to_groupby = lzip(*(ping.grouper for ping in self.groupings))
             to_groupby = Index(to_groupby)
-            return BaseGrouperGroups(self.axis.groupby(to_groupby))
+            return self.axis.groupby(to_groupby)
 
     @cache_readonly
     def is_monotonic(self):

pandas/core/indexes/base.py

@@ -4494,7 +4494,7 @@ def groupby(self, values):
         # map to the label
         result = {k: self.take(v) for k, v in compat.iteritems(result)}
 
-        return BaseGroupbyGroups(result)
+        return IndexGroupbyGroups(result)
 
     def map(self, mapper, na_action=None):
         """