simplify chunk link setter

Signed-off-by: varun-edachali-dbx <[email protected]>

Author: varun-edachali-dbx

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

@@ -130,7 +130,8 @@ def __init__(
         self._map_lock = threading.Lock()
         self.chunk_index_to_link: Dict[int, "ExternalLink"] = {}
 
-    def _set_current_chunk_link(self, link: "ExternalLink"):
+    def _set_current_chunk_link(self):
+        link = self._current_chunk_link
         with self._map_lock:
             self.chunk_index_to_link[link.chunk_index] = link