feat(polars): implement count distinct

cpcloud · cpcloud · commit aea4ccd7bc85 · 2023-03-27T12:55:24.000-04:00
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(

Original file line number	Diff line number	Diff line change
`@@ -637,6 +637,7 @@ def struct_column(op):`
`637`	`637`	`ops.StandardDev: 'std',`
`638`	`638`	`ops.Sum: 'sum',`
`639`	`639`	`ops.Variance: 'var',`
	`640`	`+ ops.CountDistinct: 'n_unique',`
`640`	`641`	`}`
`641`	`642`
`642`	`643`	`for reduction in _reductions.keys():`