fix(datafusion): handle predicates in aggregates

ibis-project · Jul 8, 2022 · 4725571 · 4725571
1 parent 9843314
commit 4725571
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/ibis/backends/datafusion/compiler.py b/ibis/backends/datafusion/compiler.py
@@ -123,6 +123,14 @@ def aggregation(op, expr):
         agg = translate(expr)
         metrics.append(agg)
 
+    if op.predicates:
+        table = table.filter(
+            functools.reduce(
+                operator.and_,
+                map(translate, op.predicates),
+            )
+        )
+
     return table.aggregate(group_by, metrics)