stricter typing for cursor

varun-edachali-dbx · varun-edachali-dbx · commit 904efe7fb7a8 · 2025-05-29T04:30:55.000Z
Signed-off-by: varun-edachali-dbx &lt;varun.edachali@databricks.com&gt;
diff --git a/src/databricks/sql/backend/databricks_client.py b/src/databricks/sql/backend/databricks_client.py
@@ -121,7 +121,7 @@ def get_query_state(self, command_id: CommandId) -> CommandState:
     def get_execution_result(
         self,
         command_id: CommandId,
-        cursor: Any,
+        cursor: "Cursor",
     ) -> "ResultSet":
         pass
 
@@ -132,7 +132,7 @@ def get_catalogs(
         session_id: SessionId,
         max_rows: int,
         max_bytes: int,
-        cursor: Any,
+        cursor: "Cursor",
     ) -> "ResultSet":
         pass
 
diff --git a/src/databricks/sql/backend/thrift_backend.py b/src/databricks/sql/backend/thrift_backend.py
@@ -1013,7 +1013,7 @@ def get_catalogs(
         session_id: SessionId,
         max_rows: int,
         max_bytes: int,
-        cursor: Any,
+        cursor: "Cursor",
     ) -> "ResultSet":
         thrift_handle = session_id.to_thrift_handle()
         if not thrift_handle:
