fix BooleanFunction

rapidsai · Dec 12, 2024 · e59f2ca · e59f2ca
1 parent c17329a
commit e59f2ca
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/python/cudf_polars/cudf_polars/dsl/expressions/boolean.py b/python/cudf_polars/cudf_polars/dsl/expressions/boolean.py
@@ -81,7 +81,16 @@ def __init__(
         self.options = options
         self.name = name
         self.children = children
-        self.is_pointwise = all(c.is_pointwise for c in self.children)
+        self.is_pointwise = self.name not in (
+            BooleanFunction.Name.All,
+            BooleanFunction.Name.AllHorizontal,
+            BooleanFunction.Name.Any,
+            BooleanFunction.Name.AnyHorizontal,
+            BooleanFunction.Name.IsDuplicated,
+            BooleanFunction.Name.IsFirstDistinct,
+            BooleanFunction.Name.IsLastDistinct,
+            BooleanFunction.Name.IsUnique,
+        ) and all(c.is_pointwise for c in self.children)
         if self.name is BooleanFunction.Name.IsIn and not all(
             c.dtype == self.children[0].dtype for c in self.children
         ):