Skip to content

Commit 9fd57d8

Browse files
set shutdown_event instead of breaking on completion so get_chunk_link is informed
Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>
1 parent 26aa45c commit 9fd57d8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/databricks/sql/backend/sea/queue.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -216,7 +216,7 @@ def _worker_loop(self):
216216
while not self._shutdown_event.is_set():
217217
links_downloaded = self._trigger_next_batch_download()
218218
if not links_downloaded:
219-
break
219+
self._shutdown_event.set()
220220
self._link_data_update.notify_all()
221221

222222
def start(self):

0 commit comments

Comments
 (0)