fix

igorbenav · May 27, 2024 · e429618 · e429618
1 parent 9947083
commit e429618
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/tests/sqlalchemy/conftest.py b/tests/sqlalchemy/conftest.py
@@ -199,11 +199,12 @@ class CardSchema(BaseModel):
 )
 
 
-async def get_session_local():
-    local_session = sessionmaker(
-        bind=async_engine, class_=AsyncSession, expire_on_commit=False
-    )
+local_session = sessionmaker(
+    bind=async_engine, class_=AsyncSession, expire_on_commit=False
+)
 
+
+async def get_session_local():
     async with local_session() as session:
         yield session
         await session.close()

diff --git a/tests/sqlmodel/conftest.py b/tests/sqlmodel/conftest.py
@@ -188,6 +188,11 @@ class CardSchema(SQLModel):
 )
 
 
+local_session = sessionmaker(
+    bind=async_engine, class_=AsyncSession, expire_on_commit=False
+)
+
+
 async def get_session_local():
     local_session = sessionmaker(bind=async_engine, class_=AsyncSession, expire_on_commit=False)
     async with local_session() as session: