Import threading

src/databricks/sql/thrift_backend.py

@@ -1,16 +1,16 @@
+from decimal import Decimal
 import errno
 import logging
 import math
-import os
 import time
 import uuid
 import threading
 from ssl import CERT_NONE, CERT_REQUIRED, create_default_context
 from typing import List, Union
 
 import pyarrow
-import thrift.protocol.TBinaryProtocol
 import thrift.transport.THttpClient
+import thrift.protocol.TBinaryProtocol
 import thrift.transport.TSocket
 import thrift.transport.TTransport
 
@@ -21,6 +21,7 @@
 from databricks.sql.auth.authenticators import AuthProvider
 from databricks.sql.thrift_api.TCLIService import TCLIService, ttypes
 from databricks.sql import *
+from databricks.sql.exc import MaxRetryDurationError
 from databricks.sql.thrift_api.TCLIService.TCLIService import (
     Client as TCLIServiceClient,
 )