diff --git a/src/triage/component/architect/builders.py b/src/triage/component/architect/builders.py index dac366f32..db5d028bd 100644 --- a/src/triage/component/architect/builders.py +++ b/src/triage/component/architect/builders.py @@ -557,7 +557,7 @@ def stitch_csvs(self, features_queries, label_query, matrix_store, matrix_uuid): logger.debug(f"getting labels pandas series from polars data frame") # getting label series - labels_pl = df_pl.select(pl.columns[-1]) + labels_pl = df_pl.select(df_pl.columns[-1]) # convert into pandas series labels_df = labels_pl.to_pandas() labels_series = labels_df.squeeze()