Make type hints match outer

timsaucer · timsaucer · commit 844592f1c268 · 2026-02-04T07:48:19.000-05:00
diff --git a/python/datafusion/user_defined.py b/python/datafusion/user_defined.py
@@ -249,7 +249,7 @@ def _function(
             )
 
         def _decorator(
-            input_fields: list[pa.DataType],
+            input_fields: Sequence[pa.DataType | pa.Field] | pa.DataType | pa.Field,
             return_field: _R,
             volatility: Volatility | str,
             name: str | None = None,

Original file line number	Diff line number	Diff line change
`@@ -249,7 +249,7 @@ def _function(`
`249`	`249`	`)`
`250`	`250`
`251`	`251`	`def _decorator(`
`252`		`- input_fields: list[pa.DataType],`
	`252`	`+ input_fields: Sequence[pa.DataType \| pa.Field] \| pa.DataType \| pa.Field,`
`253`	`253`	`return_field: _R,`
`254`	`254`	`volatility: Volatility \| str,`
`255`	`255`	`name: str \| None = None,`