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

1 files changed, 8 insertions, 10 deletions

diff --git a/lib/sqlalchemy/orm/clsregistry.py b/lib/sqlalchemy/orm/clsregistry.py
index 99a51c998..b957dc5d4 100644
--- a/lib/sqlalchemy/orm/clsregistry.py
+++ b/lib/sqlalchemy/orm/clsregistry.py
@@ -187,9 +187,9 @@ class _MultipleClassMarker(ClsRegistryToken):
         on_remove: Optional[Callable[[], None]] = None,
     ):
         self.on_remove = on_remove
-        self.contents = set(
-            [weakref.ref(item, self._remove_item) for item in classes]
-        )
+        self.contents = {
+            weakref.ref(item, self._remove_item) for item in classes
+        }
         _registries.add(self)
 
     def remove_item(self, cls: Type[Any]) -> None:
@@ -224,13 +224,11 @@ class _MultipleClassMarker(ClsRegistryToken):
         # protect against class registration race condition against
         # asynchronous garbage collection calling _remove_item,
         # [ticket:3208]
-        modules = set(
-            [
-                cls.__module__
-                for cls in [ref() for ref in self.contents]
-                if cls is not None
-            ]
-        )
+        modules = {
+            cls.__module__
+            for cls in [ref() for ref in self.contents]
+            if cls is not None
+        }
         if item.__module__ in modules:
             util.warn(
                 "This declarative base already contains a class with the "