From 91764d37c13c19b67fc292488144c47bf5c3ac6d Mon Sep 17 00:00:00 2001 From: Phillip Cloud <417981+cpcloud@users.noreply.github.com> Date: Tue, 24 Sep 2024 05:06:52 -0400 Subject: [PATCH] fix(snowflake): apply casting logic for json output to scalars (#10202) --- ibis/backends/snowflake/converter.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/ibis/backends/snowflake/converter.py b/ibis/backends/snowflake/converter.py index e3094eab94de..a4f3443e23a2 100644 --- a/ibis/backends/snowflake/converter.py +++ b/ibis/backends/snowflake/converter.py @@ -97,3 +97,9 @@ def convert_column(cls, column: pa.Array, dtype: dt.DataType) -> pa.Array: return pa.ExtensionArray.from_storage(PYARROW_JSON_TYPE, column) return super().convert_column(column, dtype) + + @classmethod + def convert_scalar(cls, scalar: pa.Scalar, dtype: dt.DataType) -> pa.Scalar: + if dtype.is_json() or dtype.is_array() or dtype.is_map() or dtype.is_struct(): + return pa.ExtensionScalar.from_storage(PYARROW_JSON_TYPE, scalar) + return super().convert_scalar(scalar, dtype)