fix: Correct column list polars materialization engine (#5595)

danbaron63 · web-flow · commit 39aeb0c443f4 · 2025-09-04T21:45:16.000-07:00
fix: returning the correct column list when requesting columnsin the PolarsBackend

Signed-off-by: Dan Baron &lt;dan.baron@starlingbank.com&gt;
diff --git a/sdk/python/feast/infra/compute_engines/local/backends/polars_backend.py b/sdk/python/feast/infra/compute_engines/local/backends/polars_backend.py
@@ -7,8 +7,8 @@
 
 
 class PolarsBackend(DataFrameBackend):
-    def columns(self, df):
-        pass
+    def columns(self, df: pl.DataFrame) -> list[str]:
+        return df.columns
 
     def from_arrow(self, table: pa.Table) -> pl.DataFrame:
         return pl.from_arrow(table)
