feat(snowflake): implement array sort

ibis-project · Aug 5, 2023 · 465fae1 · 465fae1
1 parent 4791cb4
commit 465fae1
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 11 deletions.
diff --git a/ibis/backends/snowflake/registry.py b/ibis/backends/snowflake/registry.py
@@ -459,6 +459,7 @@ def _map_get(t, op):
         ops.Median: reduction(sa.func.median),
         ops.TableColumn: _table_column,
         ops.Levenshtein: fixed_arity(sa.func.editdistance, 2),
+        ops.ArraySort: fixed_arity(sa.func.array_sort, 1),
     }
 )
 
@@ -471,7 +472,6 @@ def _map_get(t, op):
     # ibis.expr.operations.array
     ops.ArrayRemove,
     ops.ArrayRepeat,
-    ops.ArraySort,
     # ibis.expr.operations.reductions
     ops.MultiQuantile,
     # ibis.expr.operations.strings

diff --git a/ibis/backends/tests/test_array.py b/ibis/backends/tests/test_array.py
@@ -648,16 +648,7 @@ def test_array_unique(backend, con):
 
 
 @pytest.mark.notimpl(
-    [
-        "bigquery",
-        "dask",
-        "datafusion",
-        "impala",
-        "mssql",
-        "pandas",
-        "polars",
-        "snowflake",
-    ],
+    ["bigquery", "dask", "datafusion", "impala", "mssql", "pandas", "polars"],
     raises=com.OperationNotDefinedError,
 )
 @pytest.mark.notimpl(