Skip to content

Commit 318459c

Browse files
cpcloudgforsyth
authored andcommitted
refactor(snowflake): use upstream map-from-arrays function instead of a custom UDF
1 parent 42b6845 commit 318459c

File tree

2 files changed

+1
-6
lines changed

2 files changed

+1
-6
lines changed

ibis/backends/snowflake/__init__.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -95,11 +95,6 @@ class SnowflakeCompiler(AlchemyCompiler):
9595
"returns": ARRAY,
9696
"source": "return Object.values(obj)",
9797
},
98-
"ibis_udfs.public.object_from_arrays": {
99-
"inputs": {"ks": ARRAY, "vs": ARRAY},
100-
"returns": OBJECT,
101-
"source": "return Object.assign(...ks.map((k, i) => ({[k]: vs[i]})))",
102-
},
10398
"ibis_udfs.public.array_zip": {
10499
"inputs": {"arrays": ARRAY},
105100
"returns": ARRAY,

ibis/backends/snowflake/registry.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -293,7 +293,7 @@ def _timestamp_bucket(t, op):
293293
ops.Map: fixed_arity(
294294
lambda keys, values: sa.func.iff(
295295
sa.func.is_array(keys) & sa.func.is_array(values),
296-
sa.func.ibis_udfs.public.object_from_arrays(keys, values),
296+
sa.func.arrays_to_object(keys, values),
297297
sa.null(),
298298
),
299299
2,

0 commit comments

Comments
 (0)