Skip to content

Commit dfbbf79

Browse files
correct class name in logs
Signed-off-by: varun-edachali-dbx <[email protected]>
1 parent 4bd290e commit dfbbf79

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

src/databricks/sql/utils.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -281,7 +281,7 @@ def next_n_rows(self, num_rows: int) -> "pyarrow.Table":
281281
# Get remaining of num_rows or the rest of the current table, whichever is smaller
282282
length = min(num_rows, self.table.num_rows - self.table_row_index)
283283
logger.info(
284-
"SeaCloudFetchQueue: Slicing table from index {} for {} rows (table has {} rows total)".format(
284+
"CloudFetchQueue: Slicing table from index {} for {} rows (table has {} rows total)".format(
285285
self.table_row_index, length, self.table.num_rows
286286
)
287287
)
@@ -290,7 +290,7 @@ def next_n_rows(self, num_rows: int) -> "pyarrow.Table":
290290
# Concatenate results if we have any
291291
if results.num_rows > 0:
292292
logger.info(
293-
"SeaCloudFetchQueue: Concatenating {} rows to existing {} rows".format(
293+
"CloudFetchQueue: Concatenating {} rows to existing {} rows".format(
294294
table_slice.num_rows, results.num_rows
295295
)
296296
)
@@ -302,22 +302,22 @@ def next_n_rows(self, num_rows: int) -> "pyarrow.Table":
302302
rows_fetched += table_slice.num_rows
303303

304304
logger.info(
305-
"SeaCloudFetchQueue: After slice, table_row_index={}, rows_fetched={}".format(
305+
"CloudFetchQueue: After slice, table_row_index={}, rows_fetched={}".format(
306306
self.table_row_index, rows_fetched
307307
)
308308
)
309309

310310
# Replace current table with the next table if we are at the end of the current table
311311
if self.table_row_index == self.table.num_rows:
312312
logger.info(
313-
"SeaCloudFetchQueue: Reached end of current table, fetching next"
313+
"CloudFetchQueue: Reached end of current table, fetching next"
314314
)
315315
self.table = self._create_next_table()
316316
self.table_row_index = 0
317317

318318
num_rows -= table_slice.num_rows
319319

320-
logger.info("SeaCloudFetchQueue: Retrieved {} rows".format(results.num_rows))
320+
logger.info("CloudFetchQueue: Retrieved {} rows".format(results.num_rows))
321321
return results
322322

323323
def _create_empty_table(self) -> "pyarrow.Table":
@@ -330,7 +330,7 @@ def _create_table_at_offset(self, offset: int) -> Union["pyarrow.Table", None]:
330330
"""Create next table by retrieving the logical next downloaded file."""
331331
# Create next table by retrieving the logical next downloaded file, or return None to signal end of queue
332332
if not self.download_manager:
333-
logger.debug("ThriftCloudFetchQueue: No download manager available")
333+
logger.debug("CloudFetchQueue: No download manager available")
334334
return None
335335

336336
downloaded_file = self.download_manager.get_next_downloaded_file(offset)

0 commit comments

Comments
 (0)