From db8f603878d18505dd710c52581dd4a01da18e89 Mon Sep 17 00:00:00 2001 From: Nick Gerrets Date: Wed, 3 Apr 2024 11:23:55 +0200 Subject: [PATCH] numpy type --- tools/pythonpkg/src/numpy/raw_array_wrapper.cpp | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tools/pythonpkg/src/numpy/raw_array_wrapper.cpp b/tools/pythonpkg/src/numpy/raw_array_wrapper.cpp index 5d73685b20e6..7ed66929cca7 100644 --- a/tools/pythonpkg/src/numpy/raw_array_wrapper.cpp +++ b/tools/pythonpkg/src/numpy/raw_array_wrapper.cpp @@ -25,6 +25,8 @@ static idx_t GetNumpyTypeWidth(const LogicalType &type) { return sizeof(uint32_t); case LogicalTypeId::UBIGINT: return sizeof(uint64_t); + case LogicalTypeId::UHUGEINT: + return sizeof(uhugeint_t); case LogicalTypeId::TINYINT: return sizeof(int8_t); case LogicalTypeId::SMALLINT: @@ -92,6 +94,7 @@ string RawArrayWrapper::DuckDBToNumpyDtype(const LogicalType &type) { case LogicalTypeId::FLOAT: return "float32"; case LogicalTypeId::HUGEINT: + case LogicalTypeId::UHUGEINT: case LogicalTypeId::DOUBLE: case LogicalTypeId::DECIMAL: return "float64";