!squash conftest

cihai · Jun 4, 2023 · 892c70c · 892c70c
1 parent 36fd3b9
commit 892c70c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -6,6 +6,7 @@
 
 import pytest
 from sqlalchemy.orm import sessionmaker, scoped_session
+from sqlalchemy.orm.scoping import ScopedSession
 from unihan_db.bootstrap import UNIHAN_FILES
 
 
@@ -74,7 +75,7 @@ def metadata() -> sqlalchemy.MetaData:
 
 
 @pytest.fixture(scope="function")
-def session(engine, request):
+def session(engine: sqlalchemy.Engine, request: pytest.FixtureRequest) -> ScopedSession:
     connection = engine.connect()
     transaction = connection.begin()
     session_factory = sessionmaker(bind=engine)