Filter rows before selecting columns in to_polars_df

binado · Jul 25, 2024 · ff7ebbb · ff7ebbb
1 parent 973d380
commit ff7ebbb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gladeparser/backends/polars.py b/gladeparser/backends/polars.py
@@ -38,8 +38,8 @@ def to_polars_df(
         new_columns=descriptor.names,
         null_values=["null"],
         **kwargs,
-    ).select(*(cols or descriptor.names))
+    )
     if len(expr) > 0:
         query = query.filter(*expr)
 
-    return query.collect()
+    return query.select(*(cols or descriptor.names)).collect()