numpy type

nickgerrets · Apr 3, 2024 · db8f603 · db8f603
1 parent 67d27a3
commit db8f603
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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";