addeding the correct types

danbaron63 · danbaron63 · commit 0b3da5d2ccee · 2025-09-03T17:17:33.000+01:00
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,7 +7,7 @@
 
 
 class PolarsBackend(DataFrameBackend):
-    def columns(self, df):
+    def columns(self, df: pl.DataFrame) -> list[str]:
         return df.columns
 
     def from_arrow(self, table: pa.Table) -> pl.DataFrame:
