fix(snowflake): make sure pyarrow is used when possible

ibis-project · Apr 5, 2023 · 01f5154 · 01f5154
1 parent 9fb9f7f
commit 01f5154
Showing 1 changed file with 4 additions and 1 deletion.
diff --git 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 = [