fix(snowflake): use ibis-defined array_sort until upstream lands

ibis-project · Aug 5, 2023 · 6f7e13d · 6f7e13d
1 parent 76c95b2
commit 6f7e13d
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/ibis/backends/snowflake/__init__.py b/ibis/backends/snowflake/__init__.py
@@ -91,6 +91,11 @@ class SnowflakeCompiler(AlchemyCompiler):
     return Object.assign(...keys.map((key, j) => ({[key]: arrays[j][i]})));
 })""",
     },
+    "ibis_udfs.public.array_sort": {
+        "inputs": {"array": ARRAY},
+        "returns": ARRAY,
+        "source": """return array.sort();""",
+    },
 }
 
 

diff --git a/ibis/backends/snowflake/registry.py b/ibis/backends/snowflake/registry.py
@@ -461,7 +461,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),
+        ops.ArraySort: unary(sa.func.ibis_udfs.public.array_sort),
     }
 )