2323except ImportError :
2424 no_arrow_iterator_ext = True
2525
26- SQL_ENABLE_ARROW = "alter session set query_result_format='ARROW_FORCE';"
27- SQL_ENABLE_ARROW_V2 = "alter session set python_connector_query_result_format='ARROW_FORCE';"
26+ SQL_ENABLE_ARROW = "alter session set python_connector_query_result_format='ARROW';"
2827
2928EPSILON = 1e-8
3029
@@ -230,7 +229,6 @@ def test_semi_struct(conn_cnx):
230229 # fetch dataframe with new arrow support
231230 cursor_table = cnx_table .cursor ()
232231 cursor_table .execute (SQL_ENABLE_ARROW )
233- cursor_table .execute (SQL_ENABLE_ARROW_V2 )
234232 cursor_table .execute (sql_text )
235233 df_new = cursor_table .fetch_pandas_all ()
236234 col_new = df_new .iloc [0 ]
@@ -432,7 +430,6 @@ def validate_pandas(conn_cnx, sql, cases, col_count, method='one', data_type='fl
432430 # fetch dataframe with new arrow support
433431 cursor_table = cnx_table .cursor ()
434432 cursor_table .execute (SQL_ENABLE_ARROW )
435- cursor_table .execute (SQL_ENABLE_ARROW_V2 )
436433 cursor_table .execute (sql )
437434
438435 # build dataframe
@@ -527,7 +524,6 @@ def test_empty(conn_cnx):
527524 with conn_cnx () as cnx :
528525 cursor = cnx .cursor ()
529526 cursor .execute (SQL_ENABLE_ARROW )
530- cursor .execute (SQL_ENABLE_ARROW_V2 )
531527 cursor .execute ("select seq4() from table(generator(rowcount=>1)) limit 0" )
532528 assert cursor .fetch_pandas_all () is None , 'the result is not none'
533529
@@ -557,7 +553,6 @@ def fetch_pandas(conn_cnx, sql, row_count, col_count, method='one'):
557553 # fetch dataframe by fetching row by row
558554 cursor_row = cnx_row .cursor ()
559555 cursor_row .execute (SQL_ENABLE_ARROW )
560- cursor_row .execute (SQL_ENABLE_ARROW_V2 )
561556 cursor_row .execute (sql )
562557
563558 # build dataframe
@@ -583,7 +578,6 @@ def fetch_pandas(conn_cnx, sql, row_count, col_count, method='one'):
583578 # fetch dataframe with new arrow support
584579 cursor_table = cnx_table .cursor ()
585580 cursor_table .execute (SQL_ENABLE_ARROW )
586- cursor_table .execute (SQL_ENABLE_ARROW_V2 )
587581 cursor_table .execute (sql )
588582
589583 # build dataframe
0 commit comments