@@ -64,7 +64,7 @@ def __init__(
6464 """
6565
6666self .connection = connection
67- self .backend = backend # Store the backend client directly
67+ self .backend = backend
6868self .arraysize = arraysize
6969self .buffer_size_bytes = buffer_size_bytes
7070self ._next_row_index = 0
@@ -115,12 +115,12 @@ def fetchall(self) -> List[Row]:
115115pass
116116
117117@abstractmethod
118- def fetchmany_arrow (self ,size :int )-> Any :
118+ def fetchmany_arrow (self ,size :int )-> "pyarrow.Table" :
119119"""Fetch the next set of rows as an Arrow table."""
120120pass
121121
122122@abstractmethod
123- def fetchall_arrow (self )-> Any :
123+ def fetchall_arrow (self )-> "pyarrow.Table" :
124124"""Fetch all remaining rows as an Arrow table."""
125125pass
126126
@@ -207,7 +207,7 @@ def _fill_results_buffer(self):
207207use_cloud_fetch = self ._use_cloud_fetch ,
208208 )
209209self .results = results
210- self ._has_more_rows = has_more_rows
210+ self .has_more_rows = has_more_rows
211211
212212def _convert_columnar_table (self ,table ):
213213column_names = [c [0 ]for c in self .description ]
@@ -259,7 +259,7 @@ def _convert_arrow_table(self, table):
259259res = df .to_numpy (na_value = None ,dtype = "object" )
260260return [ResultRow (* v )for v in res ]
261261
262- def merge_columnar (self ,result1 ,result2 ):
262+ def merge_columnar (self ,result1 ,result2 )-> "ColumnTable" :
263263"""
264264 Function to merge / combining the columnar results into a single result
265265 :param result1:
@@ -291,7 +291,7 @@ def fetchmany_arrow(self, size: int) -> "pyarrow.Table":
291291while (
292292n_remaining_rows > 0
293293and not self .has_been_closed_server_side
294- and self ._has_more_rows
294+ and self .has_more_rows
295295 ):
296296self ._fill_results_buffer ()
297297partial_results = self .results .next_n_rows (n_remaining_rows )
@@ -316,7 +316,7 @@ def fetchmany_columnar(self, size: int):
316316while (
317317n_remaining_rows > 0
318318and not self .has_been_closed_server_side
319- and self ._has_more_rows
319+ and self .has_more_rows
320320 ):
321321self ._fill_results_buffer ()
322322partial_results = self .results .next_n_rows (n_remaining_rows )
@@ -331,7 +331,7 @@ def fetchall_arrow(self) -> "pyarrow.Table":
331331results = self .results .remaining_rows ()
332332self ._next_row_index += results .num_rows
333333
334- while not self .has_been_closed_server_side and self ._has_more_rows :
334+ while not self .has_been_closed_server_side and self .has_more_rows :
335335self ._fill_results_buffer ()
336336partial_results = self .results .remaining_rows ()
337337if isinstance (results ,ColumnTable )and isinstance (
@@ -357,7 +357,7 @@ def fetchall_columnar(self):
357357results = self .results .remaining_rows ()
358358self ._next_row_index += results .num_rows
359359
360- while not self .has_been_closed_server_side and self ._has_more_rows :
360+ while not self .has_been_closed_server_side and self .has_more_rows :
361361self ._fill_results_buffer ()
362362partial_results = self .results .remaining_rows ()
363363results = self .merge_columnar (results ,partial_results )