Fix Polars GroupBy Issue #3348

[devin-ai-integration]

Fix Polars GroupBy Issue #3348

This PR fixes an issue where group by transformations in Polars were not correctly referencing the original column names in the generated code.

Changes

• Modified the code generation in print_code.py to use pl.col() for group by columns
• Added test case test_polars_groupby_alias to verify proper column name handling in group by transformations
• Ensures both group by and aggregation operations reference original column names correctly

Testing

Added a new test that:

• Creates a test DataFrame with "group" and "age" columns
• Applies a group by transformation with max aggregation
• Verifies the transformed DataFrame structure and values
• Checks that the generated code correctly uses pl.col() for both group by and aggregation columns

Link to Devin run

https://app.devin.ai/sessions/ba11f083aa6b4f63857d6f1fbe11ac00

[devin-ai-integration]

🤖 Devin AI Engineer

I'll be helping with this pull request! Here's what you should know:

✅ I will automatically:

• Address comments on this PR. Add "(aside)" to your comment to have me ignore it.
• Look at CI failures and help fix them

⚙️ Control Options:

• Disable automatic comment and CI monitoring

[vercel]

The latest updates on your projects. Learn more about Vercel for Git ↗︎

Name	Status	Preview	Comments	Updated (UTC)
marimo-docs	✅ Ready (Inspect)	Visit Preview	💬 Add feedback	Jan 6, 2025 2:50pm
marimo-storybook	✅ Ready (Inspect)	Visit Preview	💬 Add feedback	Jan 6, 2025 2:50pm

[devin-ai-integration]

Devin is currently unreachable - the session may have died.

[github-actions]

🚀 Development release published. You may be able to view the changes at https://marimo.app?v=0.10.10-dev20