From 18db6ea6b400348132f572b43c2e5836fda71965 Mon Sep 17 00:00:00 2001 From: ahdamin Date: Wed, 9 Oct 2024 10:56:28 +0200 Subject: [PATCH] use AsyncGenerator for store instance --- genotype_api/database/store.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/genotype_api/database/store.py b/genotype_api/database/store.py index 23eaf70..56012b6 100644 --- a/genotype_api/database/store.py +++ b/genotype_api/database/store.py @@ -1,5 +1,7 @@ """Module for the store handler.""" +from typing import AsyncGenerator + from sqlalchemy.exc import OperationalError from sqlalchemy.ext.asyncio import AsyncSession from tenacity import retry, retry_if_exception_type, stop_after_attempt, wait_fixed @@ -39,7 +41,7 @@ async def create(cls) -> "Store": return cls(session) # Return a Store instance with the session -async def get_store() -> Store: +async def get_store() -> AsyncGenerator[Store, None]: """Return a Store instance.""" store = await Store.create() try: