acquite lock before notifying

Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>

Author: varun-edachali-dbx

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

@@ -277,7 +277,8 @@ def _worker_loop(self):
             if not links_downloaded:
                 self._shutdown_event.set()
         logger.debug("LinkFetcher[%s]: worker thread exiting", self._statement_id)
-        self._link_data_update.notify_all()
+        with self._link_data_update:
+            self._link_data_update.notify_all()
 
     def start(self):
         """Spawn the worker thread."""