Skip to content

Commit 233dce1

Browse files
cpcloudgforsyth
authored andcommitted
fix(duckdb): use the UDF's signature instead of arguments' output type for generating a duckdb signature
1 parent 0972f6a commit 233dce1

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

ibis/backends/duckdb/__init__.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1272,7 +1272,10 @@ def _register_udfs(self, expr: ir.Expr) -> None:
12721272
def _compile_udf(self, udf_node: ops.ScalarUDF) -> None:
12731273
func = udf_node.__func__
12741274
name = func.__name__
1275-
input_types = [DuckDBType.to_string(arg.dtype) for arg in udf_node.args]
1275+
input_types = [
1276+
DuckDBType.to_string(param.annotation.pattern.dtype)
1277+
for param in udf_node.__signature__.parameters.values()
1278+
]
12761279
output_type = DuckDBType.to_string(udf_node.dtype)
12771280

12781281
def register_udf(con):

0 commit comments

Comments
 (0)