Revert IR changes

rapidsai · Feb 14, 2025 · f5dbc59 · f5dbc59
1 parent ff252e0
commit f5dbc59
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/python/cudf_polars/cudf_polars/dsl/ir.py b/python/cudf_polars/cudf_polars/dsl/ir.py
@@ -743,9 +743,10 @@ def do_evaluate(
         projection: tuple[str, ...] | None,
     ) -> DataFrame:
         """Evaluate and return a dataframe."""
+        pdf = pl.DataFrame._from_pydf(df)
         if projection is not None:
-            df = df.select(projection)
-        df = DataFrame.from_polars(df)
+            pdf = pdf.select(projection)
+        df = DataFrame.from_polars(pdf)
         assert all(
             c.obj.type() == dtype
             for c, dtype in zip(df.columns, schema.values(), strict=True)