@@ -181,6 +181,7 @@ def __init__(
181181 :param ssl_options: SSL options for cloud fetch
182182 :param is_direct_results: Whether there are more rows to fetch
183183 """
184+
184185 # Initialize ThriftResultSet-specific attributes
185186 self ._use_cloud_fetch = use_cloud_fetch
186187 self .is_direct_results = is_direct_results
@@ -325,6 +326,7 @@ def fetchmany_arrow(self, size: int) -> "pyarrow.Table":
325326
326327 An empty sequence is returned when no more rows are available.
327328 """
329+
328330 if size < 0 :
329331 raise ValueError ("size argument for fetchmany is %s but must be >= 0" , size )
330332 results = self .results .next_n_rows (size )
@@ -349,6 +351,7 @@ def fetchmany_columnar(self, size: int):
349351 Fetch the next set of rows of a query result, returning a Columnar Table.
350352 An empty sequence is returned when no more rows are available.
351353 """
354+
352355 if size < 0 :
353356 raise ValueError ("size argument for fetchmany is %s but must be >= 0" , size )
354357
@@ -413,6 +416,7 @@ def fetchone(self) -> Optional[Row]:
413416 Fetch the next row of a query result set, returning a single sequence,
414417 or None when no more data is available.
415418 """
419+
416420 if isinstance (self .results , ColumnQueue ):
417421 res = self ._convert_columnar_table (self .fetchmany_columnar (1 ))
418422 else :
@@ -427,6 +431,7 @@ def fetchall(self) -> List[Row]:
427431 """
428432 Fetch all (remaining) rows of a query result, returning them as a list of rows.
429433 """
434+
430435 if isinstance (self .results , ColumnQueue ):
431436 return self ._convert_columnar_table (self .fetchall_columnar ())
432437 else :
@@ -438,6 +443,7 @@ def fetchmany(self, size: int) -> List[Row]:
438443
439444 An empty sequence is returned when no more rows are available.
440445 """
446+
441447 if isinstance (self .results , ColumnQueue ):
442448 return self ._convert_columnar_table (self .fetchmany_columnar (size ))
443449 else :
0 commit comments