remove string literals around forward references using from __future import annotations

Signed-off-by: varun-edachali-dbx <varun.edachali@databricks.com>

Author: varun-edachali-dbx

src/databricks/sql/backend/databricks_client.py

@@ -8,21 +8,18 @@
 - Fetching metadata about catalogs, schemas, tables, and columns
 """
 
+from __future__ import annotations
+
 from abc import ABC, abstractmethod
 from typing import Dict, Tuple, List, Optional, Any, Union, TYPE_CHECKING
 
 if TYPE_CHECKING:
     from databricks.sql.client import Cursor
+    from databricks.sql.result_set import ResultSet
 
 from databricks.sql.thrift_api.TCLIService import ttypes
 from databricks.sql.backend.types import SessionId, CommandId, CommandState
 
-# Forward reference for type hints
-from typing import TYPE_CHECKING
-
-if TYPE_CHECKING:
-    from databricks.sql.result_set import ResultSet
-
 
 class DatabricksClient(ABC):
     # == Connection and Session Management ==
@@ -80,12 +77,12 @@ def execute_command(
         max_rows: int,
         max_bytes: int,
         lz4_compression: bool,
-        cursor: "Cursor",
+        cursor: Cursor,
         use_cloud_fetch: bool,
         parameters: List[ttypes.TSparkParameter],
         async_op: bool,
         enforce_embedded_schema_correctness: bool,
-    ) -> Union["ResultSet", None]:
+    ) -> Union[ResultSet, None]:
         """
         Executes a SQL command or query within the specified session.
 
@@ -175,8 +172,8 @@ def get_query_state(self, command_id: CommandId) -> CommandState:
     def get_execution_result(
         self,
         command_id: CommandId,
-        cursor: "Cursor",
-    ) -> "ResultSet":
+        cursor: Cursor,
+    ) -> ResultSet:
         """
         Retrieves the results of a previously executed command.
 
@@ -203,8 +200,8 @@ def get_catalogs(
         session_id: SessionId,
         max_rows: int,
         max_bytes: int,
-        cursor: "Cursor",
-    ) -> "ResultSet":
+        cursor: Cursor,
+    ) -> ResultSet:
         """
         Retrieves a list of available catalogs.
 
@@ -232,10 +229,10 @@ def get_schemas(
         session_id: SessionId,
         max_rows: int,
         max_bytes: int,
-        cursor: "Cursor",
+        cursor: Cursor,
         catalog_name: Optional[str] = None,
         schema_name: Optional[str] = None,
-    ) -> "ResultSet":
+    ) -> ResultSet:
         """
         Retrieves a list of schemas, optionally filtered by catalog and schema name patterns.
 
@@ -265,12 +262,12 @@ def get_tables(
         session_id: SessionId,
         max_rows: int,
         max_bytes: int,
-        cursor: "Cursor",
+        cursor: Cursor,
         catalog_name: Optional[str] = None,
         schema_name: Optional[str] = None,
         table_name: Optional[str] = None,
         table_types: Optional[List[str]] = None,
-    ) -> "ResultSet":
+    ) -> ResultSet:
         """
         Retrieves a list of tables, optionally filtered by catalog, schema, table name, and table types.
 
@@ -302,12 +299,12 @@ def get_columns(
         session_id: SessionId,
         max_rows: int,
         max_bytes: int,
-        cursor: "Cursor",
+        cursor: Cursor,
         catalog_name: Optional[str] = None,
         schema_name: Optional[str] = None,
         table_name: Optional[str] = None,
         column_name: Optional[str] = None,
-    ) -> "ResultSet":
+    ) -> ResultSet:
         """
         Retrieves a list of columns, optionally filtered by catalog, schema, table, and column name patterns.
 

src/databricks/sql/backend/filters.py

@@ -4,6 +4,8 @@
 This module provides filtering capabilities for result sets returned by different backends.
 """
 
+from __future__ import annotations
+
 import logging
 from typing import (
     List,
@@ -37,8 +39,8 @@ class ResultSetFilter:
 
     @staticmethod
     def _filter_sea_result_set(
-        result_set: "SeaResultSet", filter_func: Callable[[List[Any]], bool]
-    ) -> "SeaResultSet":
+        result_set: SeaResultSet, filter_func: Callable[[List[Any]], bool]
+    ) -> SeaResultSet:
         """
         Filter a SEA result set using the provided filter function.
 
@@ -91,11 +93,11 @@ def _filter_sea_result_set(
 
     @staticmethod
     def filter_by_column_values(
-        result_set: "ResultSet",
+        result_set: ResultSet,
         column_index: int,
         allowed_values: List[str],
         case_sensitive: bool = False,
-    ) -> "ResultSet":
+    ) -> ResultSet:
         """
         Filter a result set by values in a specific column.
 
@@ -138,8 +140,8 @@ def filter_by_column_values(
 
     @staticmethod
     def filter_tables_by_type(
-        result_set: "ResultSet", table_types: Optional[List[str]] = None
-    ) -> "ResultSet":
+        result_set: ResultSet, table_types: Optional[List[str]] = None
+    ) -> ResultSet:
         """
         Filter a result set of tables by the specified table types.
 

src/databricks/sql/backend/sea/backend.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import logging
 import uuid
 import time
@@ -403,12 +405,12 @@ def execute_command(
         max_rows: int,
         max_bytes: int,
         lz4_compression: bool,
-        cursor: "Cursor",
+        cursor: Cursor,
         use_cloud_fetch: bool,
         parameters: List,
         async_op: bool,
         enforce_embedded_schema_correctness: bool,
-    ) -> Union["ResultSet", None]:
+    ) -> Union[ResultSet, None]:
         """
         Execute a SQL command using the SEA backend.
 
@@ -592,8 +594,8 @@ def get_query_state(self, command_id: CommandId) -> CommandState:
     def get_execution_result(
         self,
         command_id: CommandId,
-        cursor: "Cursor",
-    ) -> "ResultSet":
+        cursor: Cursor,
+    ) -> ResultSet:
         """
         Get the result of a command execution.
 
@@ -650,8 +652,8 @@ def get_catalogs(
         session_id: SessionId,
         max_rows: int,
         max_bytes: int,
-        cursor: "Cursor",
-    ) -> "ResultSet":
+        cursor: Cursor,
+    ) -> ResultSet:
         """Get available catalogs by executing 'SHOW CATALOGS'."""
         result = self.execute_command(
             operation="SHOW CATALOGS",
@@ -673,10 +675,10 @@ def get_schemas(
         session_id: SessionId,
         max_rows: int,
         max_bytes: int,
-        cursor: "Cursor",
+        cursor: Cursor,
         catalog_name: Optional[str] = None,
         schema_name: Optional[str] = None,
-    ) -> "ResultSet":
+    ) -> ResultSet:
         """Get schemas by executing 'SHOW SCHEMAS IN catalog [LIKE pattern]'."""
         if not catalog_name:
             raise ValueError("Catalog name is required for get_schemas")
@@ -706,12 +708,12 @@ def get_tables(
         session_id: SessionId,
         max_rows: int,
         max_bytes: int,
-        cursor: "Cursor",
+        cursor: Cursor,
         catalog_name: Optional[str] = None,
         schema_name: Optional[str] = None,
         table_name: Optional[str] = None,
         table_types: Optional[List[str]] = None,
-    ) -> "ResultSet":
+    ) -> ResultSet:
         """Get tables by executing 'SHOW TABLES IN catalog [SCHEMA LIKE pattern] [LIKE pattern]'."""
         if not catalog_name:
             raise ValueError("Catalog name is required for get_tables")
@@ -754,12 +756,12 @@ def get_columns(
         session_id: SessionId,
         max_rows: int,
         max_bytes: int,
-        cursor: "Cursor",
+        cursor: Cursor,
         catalog_name: Optional[str] = None,
         schema_name: Optional[str] = None,
         table_name: Optional[str] = None,
         column_name: Optional[str] = None,
-    ) -> "ResultSet":
+    ) -> ResultSet:
         """Get columns by executing 'SHOW COLUMNS IN CATALOG catalog [SCHEMA LIKE pattern] [TABLE LIKE pattern] [LIKE pattern]'."""
         if not catalog_name:
             raise ValueError("Catalog name is required for get_columns")

src/databricks/sql/result_set.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 from abc import ABC, abstractmethod
 from typing import List, Optional, Any, Union, Tuple, TYPE_CHECKING
 
@@ -21,6 +23,7 @@
 if TYPE_CHECKING:
     from databricks.sql.backend.thrift_backend import ThriftDatabricksClient
     from databricks.sql.client import Connection
+
 from databricks.sql.backend.databricks_client import DatabricksClient
 from databricks.sql.thrift_api.TCLIService import ttypes
 from databricks.sql.types import Row
@@ -40,8 +43,8 @@ class ResultSet(ABC):
 
     def __init__(
         self,
-        connection: "Connection",
-        backend: "DatabricksClient",
+        connection: Connection,
+        backend: DatabricksClient,
         arraysize: int,
         buffer_size_bytes: int,
         command_id: CommandId,
@@ -193,9 +196,9 @@ class ThriftResultSet(ResultSet):
 
     def __init__(
         self,
-        connection: "Connection",
-        execute_response: "ExecuteResponse",
-        thrift_client: "ThriftDatabricksClient",
+        connection: Connection,
+        execute_response: ExecuteResponse,
+        thrift_client: ThriftDatabricksClient,
         buffer_size_bytes: int = 104857600,
         arraysize: int = 10000,
         use_cloud_fetch: bool = True,
@@ -451,13 +454,13 @@ class SeaResultSet(ResultSet):
 
     def __init__(
         self,
-        connection: "Connection",
-        execute_response: "ExecuteResponse",
-        sea_client: "SeaDatabricksClient",
+        connection: Connection,
+        execute_response: ExecuteResponse,
+        sea_client: SeaDatabricksClient,
         buffer_size_bytes: int = 104857600,
         arraysize: int = 10000,
-        result_data: Optional["ResultData"] = None,
-        manifest: Optional["ResultManifest"] = None,
+        result_data: Optional[ResultData] = None,
+        manifest: Optional[ResultManifest] = None,
     ):
         """
         Initialize a SeaResultSet with the response from a SEA query execution.