feat: add support for pa.ExtensionType

delta-io · Sep 17, 2024 · d24ffa8 · d24ffa8
1 parent 6125adc
commit d24ffa8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/deltalake/schema.py b/python/deltalake/schema.py
@@ -95,6 +95,8 @@ def dtype_to_delta_dtype(dtype: pa.DataType) -> pa.DataType:
                 return pa.timestamp("us", "UTC")
         elif type(dtype) is pa.FixedSizeBinaryType:
             return pa.binary()
+        elif isinstance(dtype, pa.ExtensionType):
+            return dtype.storage_type
         try:
             return dtype_map[dtype]
         except KeyError: