From f7e5f884226c6e2cc7fdd8981337576b3747693a Mon Sep 17 00:00:00 2001 From: Dan Baron Date: Wed, 3 Sep 2025 17:16:35 +0100 Subject: [PATCH] fix: returning the correct column list when requesting columnsin the PolarsBackend Signed-off-by: Dan Baron --- .../infra/compute_engines/local/backends/polars_backend.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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)