fix(polars): use double explode -> implode hack to flatten

ibis-project · Sep 19, 2024 · addc6b6 · addc6b6
1 parent 9c7b1f5
commit addc6b6
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/ibis/backends/polars/compiler.py b/ibis/backends/polars/compiler.py
@@ -1021,7 +1021,11 @@ def array_flatten(op, **kw):
         .then(None)
         .when(result.list.len() == 0)
         .then([])
-        .otherwise(result.flatten())
+        # polars doesn't have a sane API (yet?) for removing one level of
+        # nesting from an array so we use this hack
+        #
+        # https://github.com/ibis-project/ibis/issues/10135
+        .otherwise(result.explode().explode().implode())
     )