From 32328f11921271f24e27e9d303af566899dbce11 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Maciek=20Bry=C5=84ski?= Date: Fri, 1 Nov 2024 19:59:19 +0100 Subject: [PATCH] Fix review comments --- alembic/ddl/impl.py | 9 +++++++-- tests/test_version_table.py | 8 +++++--- 2 files changed, 12 insertions(+), 5 deletions(-) diff --git a/alembic/ddl/impl.py b/alembic/ddl/impl.py index cc17ee75..3b1be014 100644 --- a/alembic/ddl/impl.py +++ b/alembic/ddl/impl.py @@ -140,14 +140,19 @@ def static_output(self, text: str) -> None: self.output_buffer.flush() def version_table_impl( - self, *, version_table, version_table_schema, version_table_pk, **kw + self, + *, + version_table: str, + version_table_schema: Optional[str], + version_table_pk: bool, + **kw, ) -> Table: """create the Table object for the version_table. Provided as part of impl so that third party dialects can override this. - .. versionadded:: 1.13.4 + .. versionadded:: 1.14 """ vt = Table( diff --git a/tests/test_version_table.py b/tests/test_version_table.py index 4fa36014..fb5f8193 100644 --- a/tests/test_version_table.py +++ b/tests/test_version_table.py @@ -1,3 +1,5 @@ +from typing import Optional + from sqlalchemy import Column from sqlalchemy import inspect from sqlalchemy import Integer @@ -385,9 +387,9 @@ class MyDialectImpl(impl.DefaultImpl): def version_table_impl( self, *, - version_table, - version_table_schema, - version_table_pk, + version_table: str, + version_table_schema: Optional[str], + version_table_pk: bool, **kw, ): vt = Table(