Skip to content

Conversation

@guan404ming
Copy link
Member

Related

cc @eladkal

Why

  • The original implementation converted polars DataFrames to pandas using to_pandas(), which still requires
    pandas to be installed
    .
  • No proper support for polars' native methods like with_columns() and group_by()

How

  • Implemented proper polars support by adding native polars DataFrame handling using polars-specific methods

^ Add meaningful description above
Read the Pull Request Guidelines for more information.
In case of fundamental code changes, an Airflow Improvement Proposal (AIP) is needed.
In case of a new dependency, check compliance with the ASF 3rd Party License Policy.
In case of backwards incompatible changes please leave a note in a newsfragment file, named {pr_number}.significant.rst or {issue_number}.significant.rst, in airflow-core/newsfragments.

@boring-cyborg boring-cyborg bot added area:providers provider:amazon AWS/Amazon - related issues labels Aug 17, 2025
@guan404ming guan404ming force-pushed the fix-partition-dataframe branch from fcf98ae to 57a2c59 Compare August 17, 2025 16:28
@guan404ming guan404ming force-pushed the fix-partition-dataframe branch from 57a2c59 to c810df0 Compare August 17, 2025 19:45
@guan404ming guan404ming requested a review from prdai August 17, 2025 19:45
@guan404ming guan404ming force-pushed the fix-partition-dataframe branch from c810df0 to e5658fa Compare August 19, 2025 15:12
@guan404ming guan404ming requested a review from prdai August 19, 2025 15:12
guan404ming and others added 2 commits August 22, 2025 10:51
Co-authored-by: Ranuga Disansa <79456372+Programmer-RD-AI@users.noreply.github.com>
@guan404ming guan404ming force-pushed the fix-partition-dataframe branch from e5658fa to f9abc65 Compare August 22, 2025 02:51
@potiuk potiuk merged commit d29370b into apache:main Aug 22, 2025
75 checks passed
mangal-vairalkar pushed a commit to mangal-vairalkar/airflow that referenced this pull request Aug 30, 2025
…pache#54588)

* Fix _partition_dataframe for proper polars support

* Refactor to reduce code duplicattion

Co-authored-by: Ranuga Disansa <79456372+Programmer-RD-AI@users.noreply.github.com>

---------

Co-authored-by: Ranuga Disansa <79456372+Programmer-RD-AI@users.noreply.github.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area:providers provider:amazon AWS/Amazon - related issues

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants