apache / datafusion-python

Apache DataFusion Python Bindings
https://datafusion.apache.org/python
Apache License 2.0
323 stars 64 forks source link

feat: udaf: enable multiple column input #546

Closed dlovell closed 6 months ago

dlovell commented 7 months ago

Which issue does this PR close?

This PR does not close an issue but propagates the work of https://github.com/apache/arrow-datafusion/pull/7096 to arrow-datafusion-python

Rationale for this change

The functionality already exists in the underlying rust but is not exposed to the python interface.

What changes are included in this PR?

udaf still accepts a single pa.lib.DataType as an input, but also accepts an iterable of pa.lib.DataType

Are there any user-facing changes?

All previously working code will still work the same.