Provide annotation for the session attribute

This commit is contained in:
Martijn Pieters
2023-04-27 16:32:07 +01:00
parent 9c966286b4
commit 6af14be049

View File

@@ -2,7 +2,7 @@ import datetime
from typing import Optional, Tuple
from aiobotocore.client import AioBaseClient
from aiobotocore.session import get_session
from aiobotocore.session import get_session, AioSession
from fastapi_cache.backends import Backend
@@ -26,7 +26,7 @@ class DynamoBackend(Backend):
"""
def __init__(self, table_name: str, region: Optional[str] = None) -> None:
self.session = get_session()
self.session: AioSession = get_session()
self.client: Optional[AioBaseClient] = None # Needs async init
self.table_name = table_name
self.region = region