refactor(snowflake): use upstream map-from-arrays function instead of…

… a custom UDF
ibis-project · Dec 13, 2023 · 318459c · 318459c
1 parent 42b6845
commit 318459c
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 6 deletions.
diff --git a/ibis/backends/snowflake/__init__.py b/ibis/backends/snowflake/__init__.py
@@ -95,11 +95,6 @@ class SnowflakeCompiler(AlchemyCompiler):
         "returns": ARRAY,
         "source": "return Object.values(obj)",
     },
-    "ibis_udfs.public.object_from_arrays": {
-        "inputs": {"ks": ARRAY, "vs": ARRAY},
-        "returns": OBJECT,
-        "source": "return Object.assign(...ks.map((k, i) => ({[k]: vs[i]})))",
-    },
     "ibis_udfs.public.array_zip": {
         "inputs": {"arrays": ARRAY},
         "returns": ARRAY,

diff --git a/ibis/backends/snowflake/registry.py b/ibis/backends/snowflake/registry.py
@@ -293,7 +293,7 @@ def _timestamp_bucket(t, op):
         ops.Map: fixed_arity(
             lambda keys, values: sa.func.iff(
                 sa.func.is_array(keys) & sa.func.is_array(values),
-                sa.func.ibis_udfs.public.object_from_arrays(keys, values),
+                sa.func.arrays_to_object(keys, values),
                 sa.null(),
             ),
             2,