BUG: Groupby.apply wasn't allowing for functions which return lists

doc/source/whatsnew/v1.0.1.rst

@@ -15,7 +15,7 @@ including other versions of pandas.
 
 Bug fixes
 ~~~~~~~~~
-
+- Bug in :meth:`GroupBy.apply` was raising ``TypeError`` if called with function which returned list (:issue:`31441`)
 
 Categorical
 ^^^^^^^^^^^

pandas/_libs/reduction.pyx

@@ -500,8 +500,7 @@ def apply_frame_axis0(object frame, object f, object names,
                 pass
 
             if not is_scalar(piece):
-                # Need to copy data to avoid appending references
-                if hasattr(piece, "copy"):
+                if hasattr(piece, "_typ") and piece._typ == "series":
                     piece = piece.copy(deep="all")
                 else:
                     piece = copy(piece)

pandas/tests/groupby/test_apply.py

@@ -827,3 +827,11 @@ def test_apply_index_has_complex_internals(index):
     df = DataFrame({"group": [1, 1, 2], "value": [0, 1, 0]}, index=index)
     result = df.groupby("group").apply(lambda x: x)
     tm.assert_frame_equal(result, df)
+
+
+def test_apply_function_returns_list():
+    # GH 31441
+    df = pd.DataFrame(["A", "A", "B", "B"], columns=["groups"])
+    result = df.groupby("groups").apply(lambda x: x.index.to_list())
+    expected = pd.Series([[0, 1], [2, 3]], index=pd.Index(["A", "B"], name="groups"))
+    tm.assert_series_equal(result, expected)