fix: resolve mypy errors (aclose attr-defined, dynamodb return type)

Author: wolf-j3blair

fastapi_cache/backends/dynamodb.py

@@ -46,10 +46,12 @@ async def init(self) -> None:
         ).__aenter__()
 
     async def close(self) -> None:
-        self.client = await self.client.__aexit__(None, None, None)
+        self.client = await self.client.__aexit__(None, None, None)  # type: ignore[assignment,func-returns-value]
 
     async def get_with_ttl(self, key: str) -> tuple[int, bytes | None]:
-        response = await self.client.get_item(TableName=self.table_name, Key={"key": {"S": key}})
+        response = await self.client.get_item(
+            TableName=self.table_name, Key={"key": {"S": key}}
+        )
 
         if "Item" in response:
             value = response["Item"].get("value", {}).get("B")
@@ -66,19 +68,24 @@ async def get_with_ttl(self, key: str) -> tuple[int, bytes | None]:
         return 0, None
 
     async def get(self, key: str) -> bytes | None:
-        response = await self.client.get_item(TableName=self.table_name, Key={"key": {"S": key}})
+        response = await self.client.get_item(
+            TableName=self.table_name, Key={"key": {"S": key}}
+        )
         if "Item" in response:
             return response["Item"].get("value", {}).get("B")
         return None
 
-    async def set(self, key: str, value: bytes, expire: int | None = None) -> None:
+    async def set(
+        self, key: str, value: bytes, expire: int | None = None
+    ) -> None:
         ttl = (
             {
                 "ttl": {
                     "N": str(
                         int(
                             (
-                                datetime.datetime.now() + datetime.timedelta(seconds=expire)
+                                datetime.datetime.now()
+                                + datetime.timedelta(seconds=expire)
                             ).timestamp()
                         )
                     )
@@ -99,5 +106,7 @@ async def set(self, key: str, value: bytes, expire: int | None = None) -> None:
             },
         )
 
-    async def clear(self, namespace: str | None = None, key: str | None = None) -> int:
+    async def clear(
+        self, namespace: str | None = None, key: str | None = None
+    ) -> int:
         raise NotImplementedError

tests/test_properties.py

@@ -33,7 +33,7 @@ async def test_redis_backend_roundtrip(key: str, value: bytes) -> None:
 
     assert result == value
 
-    await redis.aclose()
+    await redis.aclose()  # type: ignore[attr-defined]
 
 
 # Feature: redis5-prefect-compatibility, Property 2: TTL consistency after set
@@ -59,7 +59,7 @@ async def test_ttl_consistency_after_set(
     assert returned_ttl <= ttl
     assert returned_value == value
 
-    await redis.aclose()
+    await redis.aclose()  # type: ignore[attr-defined]
 
 
 # Feature: redis5-prefect-compatibility, Property 4: JsonCoder datetime round-trip
@@ -155,7 +155,7 @@ async def test_clear_namespace_isolation(
         result = await backend.get(key)
         assert result == other_value, f"Expected {key} to still have its value"
 
-    await redis.aclose()
+    await redis.aclose()  # type: ignore[attr-defined]
 
 
 # Feature: redis5-prefect-compatibility, Property 5: Clear with no arguments is a no-op
@@ -206,4 +206,4 @@ async def test_clear_no_args_is_noop(
             f"Expected key '{key}' to still have value {value!r}, got {stored!r}"
         )
 
-    await redis.aclose()
+    await redis.aclose()  # type: ignore[attr-defined]