summaryrefslogtreecommitdiff
path: root/lib/sqlalchemy/orm/util.py
diff options
context:
space:
mode:
authorFederico Caselli <cfederico87@gmail.com>2022-11-03 20:52:21 +0100
committerFederico Caselli <cfederico87@gmail.com>2022-11-16 23:03:04 +0100
commit4eb4ceca36c7ce931ea65ac06d6ed08bf459fc66 (patch)
tree4970cff3f78489a4a0066cd27fd4bae682402957 /lib/sqlalchemy/orm/util.py
parent3fc6c40ea77c971d3067dab0fdf57a5b5313b69b (diff)
downloadsqlalchemy-4eb4ceca36c7ce931ea65ac06d6ed08bf459fc66.tar.gz
Try running pyupgrade on the code
command run is "pyupgrade --py37-plus --keep-runtime-typing --keep-percent-format <files...>" pyupgrade will change assert_ to assertTrue. That was reverted since assertTrue does not exists in sqlalchemy fixtures Change-Id: Ie1ed2675c7b11d893d78e028aad0d1576baebb55
Diffstat (limited to 'lib/sqlalchemy/orm/util.py')
-rw-r--r--lib/sqlalchemy/orm/util.py8
1 files changed, 3 insertions, 5 deletions
diff --git a/lib/sqlalchemy/orm/util.py b/lib/sqlalchemy/orm/util.py
index 50eba5d4c..e5bdbaa4f 100644
--- a/lib/sqlalchemy/orm/util.py
+++ b/lib/sqlalchemy/orm/util.py
@@ -1011,7 +1011,7 @@ class AliasedInsp(
our_classes = util.to_set(
mp.class_ for mp in self.with_polymorphic_mappers
)
- new_classes = set([mp.class_ for mp in other.with_polymorphic_mappers])
+ new_classes = {mp.class_ for mp in other.with_polymorphic_mappers}
if our_classes == new_classes:
return other
else:
@@ -1278,8 +1278,7 @@ class LoaderCriteriaOption(CriteriaOption):
def _all_mappers(self) -> Iterator[Mapper[Any]]:
if self.entity:
- for mp_ent in self.entity.mapper.self_and_descendants:
- yield mp_ent
+ yield from self.entity.mapper.self_and_descendants
else:
assert self.root_entity
stack = list(self.root_entity.__subclasses__())
@@ -1290,8 +1289,7 @@ class LoaderCriteriaOption(CriteriaOption):
inspection.inspect(subclass, raiseerr=False),
)
if ent:
- for mp in ent.mapper.self_and_descendants:
- yield mp
+ yield from ent.mapper.self_and_descendants
else:
stack.extend(subclass.__subclasses__())