Merge pull request #8 from octoenergy/rebase-to-294

matt-fleming · web-flow · commit dc7a6cae17d0 · 2024-03-14T13:40:08.000Z
Add timeout hack
diff --git a/src/databricks/sql/thrift_backend.py b/src/databricks/sql/thrift_backend.py
@@ -206,9 +206,9 @@ def __init__(
 
         # HACK!
         timeout = THRIFT_SOCKET_TIMEOUT or kwargs.get("_socket_timeout", DEFAULT_SOCKET_TIMEOUT)
-        logger.info(f"Setting timeout HACK! to {timeout}")
-        # setTimeout defaults to 15 minutes and is expected in ms
+        logger.info(f"Setting timeout HACK! to {timeout} ms")
 
+        # setTimeout defaults to 15 minutes and is expected in ms
         self._transport.setTimeout(timeout and (float(timeout) * 1000.0))
 
         self._transport.setCustomHeaders(dict(http_headers))
