From 01f5154e76c1f24a1e2bf5b71133eecc43ab7c59 Mon Sep 17 00:00:00 2001 From: Phillip Cloud <417981+cpcloud@users.noreply.github.com> Date: Tue, 4 Apr 2023 18:17:21 -0400 Subject: [PATCH] fix(snowflake): make sure pyarrow is used when possible --- ibis/backends/snowflake/__init__.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/ibis/backends/snowflake/__init__.py b/ibis/backends/snowflake/__init__.py index c39ef1b77d66..114235043abd 100644 --- a/ibis/backends/snowflake/__init__.py +++ b/ibis/backends/snowflake/__init__.py @@ -84,7 +84,10 @@ class SnowflakeExprTranslator(AlchemyExprTranslator): class SnowflakeTableSetFormatter(_AlchemyTableSetFormatter): - def _format_in_memory_table(self, _, ref_op, translator): + def _format_in_memory_table(self, op, ref_op, translator): + if _NATIVE_ARROW: + return super()._format_in_memory_table(op, ref_op, translator) + columns = translator._schema_to_sqlalchemy_columns(ref_op.schema) rows = list(ref_op.data.to_frame().itertuples(index=False)) pos_columns = [