fix

fastapi_sqla/__init__.py

@@ -5,12 +5,12 @@
 from functools import singledispatch
 from typing import Callable, Generic, List, TypeVar, Union
 
-import sqlalchemy as sa
 import structlog
 from fastapi import Depends, FastAPI, Query, Request
 from fastapi.concurrency import contextmanager_in_threadpool
 from pydantic import BaseModel, Field
 from pydantic.generics import GenericModel
+from sqlalchemy import engine_from_config
 from sqlalchemy.ext.declarative import DeferredReflection, declarative_base
 from sqlalchemy.orm import Query as LegacyQuery
 from sqlalchemy.orm.session import Session as SqlaSession
@@ -32,7 +32,7 @@ def setup(app: FastAPI):
 
 
 def startup():
-    engine = sa.engine_from_config(os.environ, prefix="sqlalchemy_")
+    engine = engine_from_config(os.environ, prefix="sqlalchemy_")
     Base.metadata.bind = engine
     Base.prepare(engine)
     _Session.configure(bind=engine)

fastapi_sqla/_pytest_plugin.py

@@ -69,7 +69,7 @@ def sqla_reflection(sqla_modules, sqla_connection, db_url):
 @fixture(autouse=True)
 def patch_sessionmaker(db_url, sqla_connection, sqla_transaction):
     """So that all DB operations are never written to db for real."""
-    with patch("fastapi_sqla.sa.engine_from_config") as engine_from_config:
+    with patch("fastapi_sqla.engine_from_config") as engine_from_config:
         engine_from_config.return_value = sqla_connection
         yield engine_from_config