From 98120a5e2f00da9fa8b59cf8e83b4603d83f703e Mon Sep 17 00:00:00 2001 From: Lawrence Mitchell Date: Thu, 25 Jul 2024 17:26:52 +0000 Subject: [PATCH] Adapt to IR changes --- python/cudf_polars/cudf_polars/dsl/expr.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/python/cudf_polars/cudf_polars/dsl/expr.py b/python/cudf_polars/cudf_polars/dsl/expr.py index 6325feced94..01af1d3349e 100644 --- a/python/cudf_polars/cudf_polars/dsl/expr.py +++ b/python/cudf_polars/cudf_polars/dsl/expr.py @@ -885,9 +885,9 @@ def __init__( if self.name not in ( "mask_nans", "round", - "setsorted", + "set_sorted", "unique", - "dropnull", + "drop_nulls", "fill_null", ): raise NotImplementedError(f"Unary function {name=}") @@ -948,7 +948,7 @@ def do_evaluate( if maintain_order: return Column(column).sorted_like(values) return Column(column) - elif self.name == "setsorted": + elif self.name == "set_sorted": (column,) = ( child.evaluate(df, context=context, mapping=mapping) for child in self.children @@ -975,7 +975,7 @@ def do_evaluate( order=order, null_order=null_order, ) - elif self.name == "dropnull": + elif self.name == "drop_nulls": (column,) = ( child.evaluate(df, context=context, mapping=mapping) for child in self.children