diff --git a/ibis/backends/snowflake/registry.py b/ibis/backends/snowflake/registry.py index 0bef0181ceee..123e0cbb3a79 100644 --- 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 index 06c6b24487d1..529f5d401da5 100644 --- 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(