data frame columns

dssg · Nov 14, 2023 · a062b3f · a062b3f
1 parent bdcd19d
commit a062b3f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()