Fix black formatting

sqlalchemy · Oct 31, 2024 · 0934362 · 0934362
1 parent 14539e4
commit 0934362
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 7 deletions.
diff --git a/alembic/runtime/migration.py b/alembic/runtime/migration.py
@@ -199,8 +199,9 @@ def __init__(
         )
 
         self._version = self.impl.version_table_impl(
-            version_table, version_table_schema,
-            opts.get("version_table_pk", True)
+            version_table,
+            version_table_schema,
+            opts.get("version_table_pk", True),
         )
 
         log.info("Context impl %s.", self.impl.__class__.__name__)
@@ -531,8 +532,10 @@ def get_current_heads(self) -> Tuple[str, ...]:
                 return ()
         assert self.connection is not None
         return tuple(
-            row[0] for row in self.connection.execute(
-                _select(self._version.c.version_num))
+            row[0]
+            for row in self.connection.execute(
+                _select(self._version.c.version_num)
+            )
         )
 
     def _ensure_version_table(self, purge: bool = False) -> None:

diff --git a/tests/test_version_table.py b/tests/test_version_table.py
@@ -394,9 +394,7 @@ def version_table_impl(
             )
             if version_table_pk:
                 vt.append_constraint(
-                    PrimaryKeyConstraint(
-                        "id", name=f"{version_table}_pkc"
-                    )
+                    PrimaryKeyConstraint("id", name=f"{version_table}_pkc")
                 )
             return vt