Aug 22, 2025 · Aug 21, 2025
diff --git a/src/databricks/sql/utils.py b/src/databricks/sql/utils.py
            num_rows -= table_slice.num_rows

        logger.debug("CloudFetchQueue: collected {} next rows".format(results.num_rows))
        returnpyarrow.concat_tables(partial_result_chunks, use_threads=True)
        returnconcat_table_chunks(partial_result_chunks)

    def remaining_rows(self) -> "pyarrow.Table":
        """
            self.table_row_index += table_slice.num_rows
            self.table = self._create_next_table()
            self.table_row_index = 0
        returnpyarrow.concat_tables(partial_result_chunks, use_threads=True)
        returnconcat_table_chunks(partial_result_chunks)

    def _create_table_at_offset(self, offset: int) -> Union["pyarrow.Table", None]:
        """Create next table at the given row offset"""
                result_table[j].extend(table_chunks[i].column_table[j])
        return ColumnTable(result_table, table_chunks[0].column_names)
    else:
        return pyarrow.concat_tables(table_chunks, use_threads=True)
        return pyarrow.concat_tables(table_chunks)


 def build_client_context(server_hostname: str, version: str, **kwargs):
Original file line number	Diff line number	Diff line change
Expand Up		@@ -298,7 +298,7 @@ def next_n_rows(self, num_rows: int) -> "pyarrow.Table":
		num_rows -= table_slice.num_rows

		logger.debug("CloudFetchQueue: collected {} next rows".format(results.num_rows))
		returnpyarrow.concat_tables(partial_result_chunks, use_threads=True)
		returnconcat_table_chunks(partial_result_chunks)

		def remaining_rows(self) -> "pyarrow.Table":
		"""
Expand All		@@ -321,7 +321,7 @@ def remaining_rows(self) -> "pyarrow.Table":
		self.table_row_index += table_slice.num_rows
		self.table = self._create_next_table()
		self.table_row_index = 0
		returnpyarrow.concat_tables(partial_result_chunks, use_threads=True)
		returnconcat_table_chunks(partial_result_chunks)

		def _create_table_at_offset(self, offset: int) -> Union["pyarrow.Table", None]:
		"""Create next table at the given row offset"""
Expand DownExpand Up		@@ -880,7 +880,7 @@ def concat_table_chunks(
		result_table[j].extend(table_chunks[i].column_table[j])
		return ColumnTable(result_table, table_chunks[0].column_names)
		else:
		return pyarrow.concat_tables(table_chunks, use_threads=True)
		return pyarrow.concat_tables(table_chunks)


		def build_client_context(server_hostname: str, version: str, **kwargs):
Expand Down