fix: Support arro3 table schema with newer deltalake packages (#5799)

trollknurr · web-flow · commit 103c5e9477a8 · 2025-12-31T10:12:04.000+05:30
* fix: Support arro3 table schema with newer deltalake packages

Signed-off-by: Anton Shtarev &lt;a.o.shtarev@gmail.com&gt;

* Fix lint

Signed-off-by: Anton Shtarev &lt;a.o.shtarev@gmail.com&gt;

---------

Signed-off-by: Anton Shtarev &lt;a.o.shtarev@gmail.com&gt;
diff --git a/sdk/python/feast/infra/offline_stores/file_source.py b/sdk/python/feast/infra/offline_stores/file_source.py
@@ -202,11 +202,18 @@ def get_table_column_names_and_types(
                 "AWS_ENDPOINT_URL": str(self.s3_endpoint_override),
             }
 
-            schema = (
-                DeltaTable(self.path, storage_options=storage_options)
-                .schema()
-                .to_pyarrow()
-            )
+            delta_schema = DeltaTable(
+                self.path, storage_options=storage_options
+            ).schema()
+            if hasattr(delta_schema, "to_arrow"):
+                # deltalake >= 0.10.0
+                arro3_schema = delta_schema.to_arrow()
+                schema = pyarrow.schema(arro3_schema)
+            elif hasattr(delta_schema, "to_pyarrow"):
+                # deltalake < 0.10.0
+                schema = delta_schema.to_pyarrow()
+            else:
+                raise Exception("Unknown DeltaTable package version")
         else:
             raise Exception(f"Unknown FileFormat -> {self.file_format}")
 
