Support all column types in SignalSchema.from_column_types

iterative · Aug 19, 2024 · c5bbb39 · c5bbb39
1 parent 61aeed4
commit c5bbb39
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/datachain/lib/convert/flatten.py b/src/datachain/lib/convert/flatten.py
@@ -10,19 +10,25 @@
     Boolean,
     DateTime,
     Float,
+    Float32,
+    Float64,
     Int,
     Int32,
     Int64,
     NullType,
     String,
+    UInt64,
 )
 
 DATACHAIN_TO_TYPE = {
     Int: int,
     Int32: int,
     Int64: int,
+    UInt64: int,
     String: str,
     Float: float,
+    Float32: float,
+    Float64: float,
     Boolean: bool,
     DateTime: datetime,
     Binary: bytes,