From a062b3f3599cb857035737aeb050e7adffc665b0 Mon Sep 17 00:00:00 2001 From: Liliana Millan Date: Tue, 14 Nov 2023 01:07:22 +0000 Subject: [PATCH] data frame columns --- src/triage/component/architect/builders.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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()