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: