diff --git a/graphene_sqlalchemy/batching.py b/graphene_sqlalchemy/batching.py index a580451..731d764 100644 --- a/graphene_sqlalchemy/batching.py +++ b/graphene_sqlalchemy/batching.py @@ -137,9 +137,7 @@ def _get_loader(relationship_prop): RELATIONSHIP_LOADERS_CACHE[relationship_prop] = loader return loader - loader = _get_loader(relationship_prop) - async def resolve(root, info, **args): - return await loader.load(root) + return await _get_loader(relationship_prop).load(root) return resolve diff --git a/graphene_sqlalchemy/tests/models.py b/graphene_sqlalchemy/tests/models.py index b638b5d..45d2379 100644 --- a/graphene_sqlalchemy/tests/models.py +++ b/graphene_sqlalchemy/tests/models.py @@ -26,7 +26,6 @@ from graphene_sqlalchemy.utils import SQL_VERSION_HIGHER_EQUAL_THAN_1_4, SQL_VERSION_HIGHER_EQUAL_THAN_2 # fmt: off -import sqlalchemy if SQL_VERSION_HIGHER_EQUAL_THAN_2: from sqlalchemy.sql.sqltypes import HasExpressionLookup # noqa # isort:skip else: