This is an automated email from the ASF dual-hosted git repository.
timsaucer pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/datafusion-python.git
The following commit(s) were added to refs/heads/main by this push:
new baec559b fix: type scalar UDF returns as Arrow arrays (#1528)
baec559b is described below
commit baec559b0a7c85934338d6da80ffbe538004f4d4
Author: BharatDeva <[email protected]>
AuthorDate: Thu May 28 14:28:47 2026 -0500
fix: type scalar UDF returns as Arrow arrays (#1528)
Co-authored-by: BharatDeva <[email protected]>
---
python/datafusion/user_defined.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/python/datafusion/user_defined.py
b/python/datafusion/user_defined.py
index a8ee7756..ced707a9 100644
--- a/python/datafusion/user_defined.py
+++ b/python/datafusion/user_defined.py
@@ -33,7 +33,7 @@ from datafusion.expr import Expr
if TYPE_CHECKING:
from _typeshed import CapsuleType as _PyCapsule
- _R = TypeVar("_R", bound=pa.DataType)
+ _R = TypeVar("_R", bound=pa.Array)
from collections.abc import Callable, Sequence
@@ -137,7 +137,7 @@ class ScalarUDF:
name: str,
func: Callable[..., _R],
input_fields: list[pa.Field],
- return_field: _R,
+ return_field: pa.Field,
volatility: Volatility | str,
) -> None:
"""Instantiate a scalar user-defined function (UDF).
@@ -311,7 +311,7 @@ class ScalarUDF:
def _decorator(
input_fields: Sequence[pa.DataType | pa.Field] | pa.DataType |
pa.Field,
- return_field: _R,
+ return_field: pa.DataType | pa.Field,
volatility: Volatility | str,
name: str | None = None,
) -> Callable:
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
