fix type check error

Signed-off-by: Nikolay Ulmasov <[email protected]>
delta-io · Nov 18, 2023 · bf73969 · bf73969
1 parent 2d67292
commit bf73969
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/python/deltalake/writer.py b/python/deltalake/writer.py
@@ -238,15 +238,20 @@ def _large_to_normal_dtype(dtype: pa.DataType) -> pa.DataType:
             return dtype
 
     if partition_by:
+        table_schema: pa.Schema = schema
         if PYARROW_MAJOR_VERSION < 12:
             partition_schema = pa.schema(
                 [
-                    pa.field(name, _large_to_normal_dtype(schema.field(name).type))
+                    pa.field(
+                        name, _large_to_normal_dtype(table_schema.field(name).type)
+                    )
                     for name in partition_by
                 ]
             )
         else:
-            partition_schema = pa.schema([schema.field(name) for name in partition_by])
+            partition_schema = pa.schema(
+                [table_schema.field(name) for name in partition_by]
+            )
         partitioning = ds.partitioning(partition_schema, flavor="hive")
     else:
         partitioning = None