feat(snowflake): any/all/notany/notall reductions

ibis-project · Jan 19, 2023 · ba1af5e · ba1af5e
1 parent b606e3d
commit ba1af5e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/ibis/backends/snowflake/registry.py b/ibis/backends/snowflake/registry.py
@@ -217,6 +217,10 @@ def _map(_, op):
         ops.ArrayCollect: reduction(sa.func.array_agg),
         ops.StringSplit: fixed_arity(sa.func.split, 2),
         ops.Map: _map,
+        ops.All: reduction(sa.func.booland_agg),
+        ops.NotAll: reduction(lambda arg: ~sa.func.booland_agg(arg)),
+        ops.Any: reduction(sa.func.boolor_agg),
+        ops.NotAny: reduction(lambda arg: ~sa.func.boolor_agg(arg)),
     }
 )
 
@@ -231,14 +235,10 @@ def _map(_, op):
     ops.ArrayRepeat,
     ops.Unnest,
     # ibis.expr.operations.reductions
-    ops.All,
-    ops.Any,
     ops.BitAnd,
     ops.BitOr,
     ops.BitXor,
     ops.MultiQuantile,
-    ops.NotAll,
-    ops.NotAny,
     # ibis.expr.operations.strings
     ops.FindInSet,
     ops.RegexExtract,