summaryrefslogtreecommitdiff
path: root/lib/sqlalchemy/ext/mypy/decl_class.py
diff options
context:
space:
mode:
authormike bayer <mike_mp@zzzcomputing.com>2021-08-25 18:23:08 +0000
committerGerrit Code Review <gerrit@ci3.zzzcomputing.com>2021-08-25 18:23:08 +0000
commite2d9ef3fe6f7bd9b151caf71ae5eb7f15522ec8c (patch)
treec718cb19b990a4fd1b9a27dc60091e71079e82ba /lib/sqlalchemy/ext/mypy/decl_class.py
parentf5b3103ac2798fb2e5d15541b1489d0d13edf4e7 (diff)
parent07b7e1594deecc6c9c64893a63f56403da0dcf05 (diff)
downloadsqlalchemy-e2d9ef3fe6f7bd9b151caf71ae5eb7f15522ec8c.tar.gz
Merge "ensure "sqlalchemy" info set for all considered classes"
Diffstat (limited to 'lib/sqlalchemy/ext/mypy/decl_class.py')
-rw-r--r--lib/sqlalchemy/ext/mypy/decl_class.py3
1 files changed, 3 insertions, 0 deletions
diff --git a/lib/sqlalchemy/ext/mypy/decl_class.py b/lib/sqlalchemy/ext/mypy/decl_class.py
index 23c78aa51..b85ec0f69 100644
--- a/lib/sqlalchemy/ext/mypy/decl_class.py
+++ b/lib/sqlalchemy/ext/mypy/decl_class.py
@@ -61,6 +61,9 @@ def scan_declarative_assignments_and_apply_types(
List[util.SQLAlchemyAttribute]
] = util.get_mapped_attributes(info, api)
+ # used by assign.add_additional_orm_attributes among others
+ util.establish_as_sqlalchemy(info)
+
if mapped_attributes is not None:
# ensure that a class that's mapped is always picked up by
# its mapped() decorator or declarative metaclass before