Add future prove args

sqlalchemy · Oct 31, 2024 · d5b8196 · d5b8196
1 parent f0afc12
commit d5b8196
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/alembic/ddl/impl.py b/alembic/ddl/impl.py
@@ -140,7 +140,7 @@ def static_output(self, text: str) -> None:
         self.output_buffer.flush()
 
     def version_table_impl(
-        self, version_table, version_table_schema, version_table_pk
+        self, *, version_table, version_table_schema, version_table_pk, **kw
     ) -> Table:
         """create the Table object for the version_table.
 

diff --git a/alembic/runtime/migration.py b/alembic/runtime/migration.py
@@ -199,9 +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,
+            version_table_schema=version_table_schema,
+            version_table_pk=opts.get("version_table_pk", True),
         )
 
         log.info("Context impl %s.", self.impl.__class__.__name__)