diff options
Diffstat (limited to 'lib/sqlalchemy/ext/assignmapper.py')
| -rw-r--r-- | lib/sqlalchemy/ext/assignmapper.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/lib/sqlalchemy/ext/assignmapper.py b/lib/sqlalchemy/ext/assignmapper.py index 5b77052e2..e672d835e 100644 --- a/lib/sqlalchemy/ext/assignmapper.py +++ b/lib/sqlalchemy/ext/assignmapper.py @@ -1,9 +1,9 @@ -from sqlalchemy import mapper, util +from sqlalchemy import mapper, util, Query import types def monkeypatch_query_method(ctx, class_, name): def do(self, *args, **kwargs): - query = class_.mapper.query(session=ctx.current) + query = Query(class_, session=ctx.current) return getattr(query, name)(*args, **kwargs) setattr(class_, name, classmethod(do)) |