From f82b0b41fbb40dcdead991f87fbd48489bb1ee59 Mon Sep 17 00:00:00 2001 From: Afroz Alam Date: Thu, 1 Feb 2024 13:49:20 -0800 Subject: [PATCH] use split_blocks=True by default --- src/snowflake/snowpark/_internal/server_connection.py | 4 ++-- src/snowflake/snowpark/mock/_connection.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/src/snowflake/snowpark/_internal/server_connection.py b/src/snowflake/snowpark/_internal/server_connection.py index a442d724553..402ee6b63aa 100644 --- a/src/snowflake/snowpark/_internal/server_connection.py +++ b/src/snowflake/snowpark/_internal/server_connection.py @@ -443,11 +443,11 @@ def _to_data_or_iter( functools.partial( _fix_pandas_df_fixed_type, results_cursor=results_cursor ), - results_cursor.fetch_pandas_batches(), + results_cursor.fetch_pandas_batches(split_blocks=True), ) if to_iter else _fix_pandas_df_fixed_type( - results_cursor.fetch_pandas_all(), results_cursor + results_cursor.fetch_pandas_all(split_blocks=True), results_cursor ) ) except NotSupportedError: diff --git a/src/snowflake/snowpark/mock/_connection.py b/src/snowflake/snowpark/mock/_connection.py index 27c54c166c6..2968131cb34 100644 --- a/src/snowflake/snowpark/mock/_connection.py +++ b/src/snowflake/snowpark/mock/_connection.py @@ -339,11 +339,11 @@ def _to_data_or_iter( functools.partial( _fix_pandas_df_fixed_type, results_cursor=results_cursor ), - results_cursor.fetch_pandas_batches(), + results_cursor.fetch_pandas_batches(split_blocks=True), ) if to_iter else _fix_pandas_df_fixed_type( - results_cursor.fetch_pandas_all(), results_cursor + results_cursor.fetch_pandas_all(split_blocks=True), results_cursor ) ) except NotSupportedError: