feat(polars): array sort

ibis-project · Aug 9, 2024 · 9a2563b · 9a2563b
1 parent ca85ae2
commit 9a2563b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/ibis/backends/polars/compiler.py b/ibis/backends/polars/compiler.py
@@ -955,6 +955,12 @@ def timestamp_diff(op, **kw):
     return left.dt.truncate("1s") - right.dt.truncate("1s")
 
 
+@translate.register(ops.ArraySort)
+def array_sort(op, **kw):
+    arg = translate(op.arg, **kw)
+    return arg.list.sort()
+
+
 @translate.register(ops.ArrayLength)
 def array_length(op, **kw):
     arg = translate(op.arg, **kw)

diff --git a/ibis/backends/tests/test_array.py b/ibis/backends/tests/test_array.py
@@ -718,7 +718,6 @@ def test_array_unique(con, input, expected):
 
 
 @builtin_array
-@pytest.mark.notimpl(["polars"], raises=com.OperationNotDefinedError)
 @pytest.mark.notyet(
     ["risingwave"],
     raises=AssertionError,