1 files changed, 12 insertions, 5 deletions

diff --git a/db/session.py b/db/session.py
index 82c0211..48ef3b6 100644
--- a/db/session.py
+++ b/db/session.py
@@ -1,21 +1,28 @@
 from contextlib import contextmanager
 
-from sqlalchemy.orm import sessionmaker, scoped_session
-from sqlmodel import create_engine
-from sqlmodel import Session
+from sqlalchemy.orm import scoped_session, sessionmaker
+from sqlmodel import Session, create_engine
 
 from conf import DATABASE_URL
 from utils.json import json_serializer
 
-
-engine = create_engine(DATABASE_URL, echo=True, future=True, json_serializer=json_serializer, pool_size=100, pool_recycle=3600)
+engine = create_engine(
+    DATABASE_URL,
+    echo=True,
+    future=True,
+    json_serializer=json_serializer,
+    pool_size=100,
+    pool_recycle=3600,
+)
 session_factory = sessionmaker(engine, class_=Session)
 ScopedSession = scoped_session(session_factory=session_factory)
 
+
 def get_session():
     with ScopedSession() as session:
         yield session
 
+
 @contextmanager
 def get_session_context(autocommit=True):
     session = next(get_session())