feat(polars): implement array flatten support

ibis-project · Nov 21, 2023 · 19b2aa0 · 19b2aa0
1 parent 3cbed71
commit 19b2aa0
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/ibis/backends/polars/compiler.py b/ibis/backends/polars/compiler.py
@@ -885,6 +885,11 @@ def array_collect(op, **kw):
     return arg
 
 
+@translate.register(ops.ArrayFlatten)
+def array_flatten(op, **kw):
+    return pl.concat_list(translate(op.arg, **kw))
+
+
 _date_methods = {
     ops.ExtractDay: "day",
     ops.ExtractMonth: "month",