From 318459cd54214bcda548c43d1d2425ec39a890d2 Mon Sep 17 00:00:00 2001 From: Phillip Cloud <417981+cpcloud@users.noreply.github.com> Date: Wed, 13 Dec 2023 11:57:12 -0500 Subject: [PATCH] refactor(snowflake): use upstream map-from-arrays function instead of a custom UDF --- ibis/backends/snowflake/__init__.py | 5 ----- ibis/backends/snowflake/registry.py | 2 +- 2 files changed, 1 insertion(+), 6 deletions(-) diff --git a/ibis/backends/snowflake/__init__.py b/ibis/backends/snowflake/__init__.py index d5a3a7ca0161..e263ce30a33f 100644 --- 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 index 7aa4b2012b69..31a9db4cbfa7 100644 --- 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,