Skip to content

Commit 2a5e3ed

Browse files
authored
fix(duckdb): support version >= 1.4.1 (#1377)
1 parent ad728d0 commit 2a5e3ed

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

python/xorq/vendor/ibis/backends/duckdb/__init__.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,9 +46,15 @@
4646
from xorq.vendor.ibis.expr.schema import SchemaLike
4747

4848

49+
try:
50+
from duckdb import func as _duckdb_func
51+
except ImportError:
52+
from duckdb import functional as _duckdb_func
53+
54+
4955
_UDF_INPUT_TYPE_MAPPING = {
50-
InputType.PYARROW: duckdb.functional.ARROW,
51-
InputType.PYTHON: duckdb.functional.NATIVE,
56+
InputType.PYARROW: _duckdb_func.ARROW,
57+
InputType.PYTHON: _duckdb_func.NATIVE,
5258
}
5359

5460

0 commit comments

Comments
 (0)