feat(polars): implement count distinct

ibis-project · Mar 27, 2023 · aea4ccd · aea4ccd
1 parent 60e7731
commit aea4ccd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/ibis/backends/polars/compiler.py b/ibis/backends/polars/compiler.py
@@ -637,6 +637,7 @@ def struct_column(op):
     ops.StandardDev: 'std',
     ops.Sum: 'sum',
     ops.Variance: 'var',
+    ops.CountDistinct: 'n_unique',
 }
 
 for reduction in _reductions.keys():

diff --git a/ibis/backends/tests/test_aggregation.py b/ibis/backends/tests/test_aggregation.py
@@ -282,7 +282,7 @@ def mean_and_std(v):
             lambda t, where: t.bool_col[where].dropna().nunique(),
             id='nunique',
             marks=pytest.mark.notimpl(
-                ["polars", "pyspark", "datafusion"], raises=com.OperationNotDefinedError
+                ["pyspark", "datafusion"], raises=com.OperationNotDefinedError
             ),
         ),
         param(