fix(client): satisfy ruff format and pyright

Author: cchinchilla-dev

src/a2a/client/transports/retry.py

@@ -45,10 +45,12 @@
     import grpc as _grpc
 
     _AioRpcError: Any = _grpc.aio.AioRpcError
-    _RETRYABLE_GRPC_CODES: frozenset[Any] = frozenset({
-        _grpc.StatusCode.UNAVAILABLE,
-        _grpc.StatusCode.RESOURCE_EXHAUSTED,
-    })
+    _RETRYABLE_GRPC_CODES: frozenset[Any] = frozenset(
+        {
+            _grpc.StatusCode.UNAVAILABLE,
+            _grpc.StatusCode.RESOURCE_EXHAUSTED,
+        }
+    )
 except ImportError:
     _AioRpcError = None
     _RETRYABLE_GRPC_CODES = frozenset()
@@ -79,7 +81,7 @@ def default_retry_predicate(error: Exception) -> bool:  # noqa: PLR0911
     if isinstance(cause, httpx.RequestError):
         return True
     if _AioRpcError is not None and isinstance(cause, _AioRpcError):
-        return cause.code() in _RETRYABLE_GRPC_CODES
+        return cause.code() in _RETRYABLE_GRPC_CODES  # pyright: ignore[reportAttributeAccessIssue]
     return False
 
 

tests/client/transports/test_retry.py

@@ -632,10 +632,13 @@ async def test_cancelled_error_during_sleep_propagates(
         async def cancelling_sleep(*_args: object, **_kwargs: object) -> None:
             raise asyncio.CancelledError
 
-        with patch(
-            'a2a.client.transports.retry.asyncio.sleep',
-            side_effect=cancelling_sleep,
-        ), pytest.raises(asyncio.CancelledError):
+        with (
+            patch(
+                'a2a.client.transports.retry.asyncio.sleep',
+                side_effect=cancelling_sleep,
+            ),
+            pytest.raises(asyncio.CancelledError),
+        ):
             await transport.send_message(SendMessageRequest())
 
         # First attempt ran; cancel hit on the sleep before the second.