expect session_id in protocol version getter

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

Author: varun-edachali-dbx

src/databricks/sql/client.py

@@ -47,7 +47,7 @@
 from databricks.sql.auth.auth import get_python_sql_connector_auth_provider
 from databricks.sql.experimental.oauth_persistence import OAuthPersistence
 from databricks.sql.session import Session
-from databricks.sql.backend.types import CommandId, BackendType
+from databricks.sql.backend.types import CommandId, BackendType, SessionId
 
 from databricks.sql.thrift_api.TCLIService.ttypes import (
     TSparkParameter,
@@ -325,9 +325,9 @@ def protocol_version(self):
         return self.session.protocol_version
 
     @staticmethod
-    def get_protocol_version(openSessionResp):
+    def get_protocol_version(session_id: SessionId):
         """Delegate to Session class static method"""
-        return Session.get_protocol_version(openSessionResp)
+        return Session.get_protocol_version(session_id)
 
     @property
     def open(self) -> bool:
@@ -868,19 +868,6 @@ def is_query_pending(self):
             ttypes.TOperationState.PENDING_STATE,
         ]
 
-    def is_query_pending(self):
-        """
-        Checks whether the async executing query is in pending state or not
-
-        :return:
-        """
-        operation_state = self.get_query_state()
-
-        return not operation_state or operation_state in [
-            ttypes.TOperationState.RUNNING_STATE,
-            ttypes.TOperationState.PENDING_STATE,
-        ]
-
     def get_async_execution_result(self):
         """