chore: lint

Author: jackwotherspoon

google/cloud/sql/connector/connector.py

@@ -65,7 +65,7 @@ def __init__(
         user_agent: Optional[str] = None,
         universe_domain: Optional[str] = None,
         refresh_strategy: str | RefreshStrategy = RefreshStrategy.BACKGROUND,
-        resolver: DefaultResolver | DnsResolver = DefaultResolver,
+        resolver: Type[DefaultResolver] | Type[DnsResolver] = DefaultResolver,
     ) -> None:
         """Initializes a Connector instance.
 
@@ -107,6 +107,7 @@ def __init__(
                 of the following: RefreshStrategy.LAZY ("LAZY") or
                 RefreshStrategy.BACKGROUND ("BACKGROUND").
                 Default: RefreshStrategy.BACKGROUND
+
             resolver (DefaultResolver | DnsResolver): The class name of the
                 resolver to use for resolving the Cloud SQL instance connection
                 name. To resolve a DNS record to an instance connection name, use

google/cloud/sql/connector/resolver.py

@@ -15,13 +15,14 @@
 from dns.asyncresolver import Resolver
 
 from google.cloud.sql.connector.connection_name import _parse_instance_connection_name
+from google.cloud.sql.connector.connection_name import ConnectionName
 from google.cloud.sql.connector.exceptions import DnsResolutionError
 
 
 class DefaultResolver:
     """DefaultResolver simply validates and parses instance connection name."""
 
-    async def resolve(self, connection_name: str) -> str:
+    async def resolve(self, connection_name: str) -> ConnectionName:
         return _parse_instance_connection_name(connection_name)
 
 
@@ -31,7 +32,7 @@ class DnsResolver(Resolver):
     TXT records in DNS.
     """
 
-    async def resolve(self, dns: str) -> str:
+    async def resolve(self, dns: str) -> ConnectionName:  # type: ignore
         try:
             conn_name = _parse_instance_connection_name(dns)
         except ValueError: