diff --git a/google/cloud/alloydbconnector/async_connector.py b/google/cloud/alloydbconnector/async_connector.py index 96c265c8..d834f262 100644 --- a/google/cloud/alloydbconnector/async_connector.py +++ b/google/cloud/alloydbconnector/async_connector.py @@ -252,7 +252,7 @@ async def _remove_cached(self, instance_uri: str) -> None: cache = self._cache.pop(instance_uri) await cache.close() - async def __aenter__(self) -> Any: + async def __aenter__(self) -> AsyncConnector: """Enter async context manager by returning Connector object""" return self diff --git a/google/cloud/alloydbconnector/psycopg.py b/google/cloud/alloydbconnector/psycopg.py index 8e91c076..bf25d927 100644 --- a/google/cloud/alloydbconnector/psycopg.py +++ b/google/cloud/alloydbconnector/psycopg.py @@ -43,7 +43,7 @@ def _proxy(local: socket.socket, remote: "ssl.SSLSocket") -> None: remote: The SSL socket connected to the AlloyDB proxy server. """ - def forward(src: Any, dst: Any) -> None: + def forward(src: socket.socket, dst: socket.socket) -> None: buf = bytearray(_CHUNK_SIZE) view = memoryview(buf) try: diff --git a/pyproject.toml b/pyproject.toml index bf2ede8f..b01657d1 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -115,11 +115,8 @@ select = [ "ANN", # flake8-annotations ] ignore = [ - "E203", - "E231", - "E266", - "E501", - "ANN401", + "ANN401", # Any is needed for pass-through **kwargs + "E501", # line too long — enforced by ruff format ] [tool.ruff.lint.per-file-ignores]