Fix groupby(as_index=False).size not reseting index

python/cudf/cudf/core/groupby/groupby.py

@@ -497,11 +497,14 @@ def size(self):
         col = cudf.core.column.column_empty(
             len(self.obj), "int8", masked=False
         )
-        return (
-            cudf.Series._from_column(col)
+        result = (
+            cudf.Series._from_column(col, name=getattr(self.obj, "name", None))
             .groupby(self.grouping, sort=self._sort, dropna=self._dropna)
             .agg("size")
         )
+        if not self._as_index:
+            result = result.rename("size").reset_index()
+        return result
 
     @_performance_tracking
     def cumcount(self, ascending: bool = True):
@@ -803,7 +806,6 @@ def agg(self, func=None, *args, engine=None, engine_kwargs=None, **kwargs):
                         right_order, len(result), nullify=False
                     )
                 )
-
         if not self._as_index:
             result = result.reset_index()
         if libgroupby._is_all_scan_aggregate(normalized_aggs):

python/cudf/cudf/tests/test_groupby.py

@@ -4074,3 +4074,17 @@ def test_get_group_list_like():
 
     with pytest.raises(KeyError):
         df.groupby(["a"]).get_group([1])
+
+
+def test_size_as_index_false():
+    df = pd.DataFrame({"a": [1, 2, 1], "b": [1, 2, 3]}, columns=["a", "b"])
+    expected = df.groupby("a", as_index=False).size()
+    result = cudf.from_pandas(df).groupby("a", as_index=False).size()
+    assert_eq(result, expected)
+
+
+def test_size_series_with_name():
+    ser = pd.Series(range(3), name="foo")
+    expected = ser.groupby(ser).size()
+    result = cudf.from_pandas(ser).groupby(ser).size()
+    assert_eq(result, expected)