Skip to content

Commit b765e33

Browse files
clean up Queue more
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent 961873a commit b765e33

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/databricks/sql/cloud_fetch_queue.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -418,7 +418,7 @@ def _create_next_table(self) -> Union["pyarrow.Table", None]:
418418
if self._current_chunk_link is None
419419
else self._current_chunk_link.next_chunk_index
420420
)
421-
if next_chunk_index is None: # This can happen if we're at the end of chunks
421+
if next_chunk_index is None:
422422
logger.debug(
423423
"SeaCloudFetchQueue: Reached end of chunks (next_chunk_index is None)"
424424
)
@@ -432,7 +432,6 @@ def _create_next_table(self) -> Union["pyarrow.Table", None]:
432432

433433
# Update current chunk to the next one
434434
self._current_chunk_index = next_chunk_index
435-
self._current_chunk_link = None
436435
try:
437436
self._current_chunk_link = self._fetch_chunk_link(next_chunk_index)
438437
except Exception as e:

0 commit comments

Comments
 (0)