[BEAM-12444] GroupBy.apply on a series grouped by a callable fails

• Type: Bug
• Status: Resolved
• Priority: P2
• Resolution: Fixed
• Affects Version/s: 2.29.0, 2.30.0, 2.31.0
• Fix Version/s: 2.32.0
• Component/s: dsl-dataframe, sdk-py-core
• Labels: dataframe-api
• Assignee: Brian Hulette
• Reporter: Brian Hulette
• Created: 02/Jun/21 17:04
• Updated: 22/Jun/21 19:22
• Resolved: 11/Jun/21 15:25

Description

df.foo.groupby(lambda x: x % 3).apply(...)

This example fails because it tries to call set_index on a Series. It fails at construction time, not at execution time.

To workaround users can call groupby on a single column dataframe instead, e.g. df[['foo']].groupby...

Issue Links

• links to: GitHub Pull Request #14930