reverting change in close in telemetry client

Signed-off-by: Sai Shree Pradhan <saishree.pradhan@databricks.com>

Author: saishreeeee

src/databricks/sql/client.py

@@ -470,7 +470,7 @@ def _close(self, close_cursors=True) -> None:
 
         self.open = False
 
-        TelemetryClientFactory.close(self.get_session_id_hex())
+        self._telemetry_client.close()
 
     def commit(self):
         """No-op because Databricks does not support transactions"""

src/databricks/sql/telemetry/telemetry_client.py

@@ -300,6 +300,7 @@ def close(self):
         """Flush remaining events before closing"""
         logger.debug("Closing TelemetryClient for connection %s", self._connection_uuid)
         self._flush()
+        TelemetryClientFactory.close(self._connection_uuid)
 
 
 class TelemetryClientFactory:
@@ -415,7 +416,6 @@ def close(connection_uuid):
                 logger.debug(
                     "Removing telemetry client for connection %s", connection_uuid
                 )
-                TelemetryClientFactory.get_telemetry_client(connection_uuid).close()
                 TelemetryClientFactory._clients.pop(connection_uuid, None)
 
             # Shutdown executor if no more clients