chore: recreate loader if old loader is on different loop (#395)

* Recreate loader if old loader is on incorrect loop * Lint --------- Co-authored-by: Cadu <[email protected]>
graphql-python · Oct 9, 2023 · b94230e · b94230e
1 parent 1436807
commit b94230e
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 4 deletions.
diff --git 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
@@ -27,7 +27,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: