fix: import CommandId in client tests

Author: varun-edachali-dbx

src/databricks/sql/auth/retry.py

@@ -19,7 +19,6 @@
 from urllib3.util.retry import RequestHistory
 
 
-
 from databricks.sql.exc import (
     CursorAlreadyClosedError,
     MaxRetryDurationError,

tests/unit/test_client.py

@@ -23,6 +23,7 @@
 from databricks.sql.exc import RequestError, CursorAlreadyClosedError
 from databricks.sql.types import Row
 from databricks.sql.result_set import ResultSet, ThriftResultSet
+from databricks.sql.backend.types import CommandId
 
 from tests.unit.test_fetches import FetchTests
 from tests.unit.test_thrift_backend import ThriftBackendTestSuite