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
index 5c874aa4433..352ffecdab8 100644
--- 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)