Fixes: #1348 Typings for target_metadata sequence of metadatas

sqlalchemy · Nov 8, 2023 · 2f68033 · 2f68033
1 parent 0e0131b
commit 2f68033
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/alembic/context.pyi b/alembic/context.pyi
@@ -14,6 +14,7 @@ from typing import Mapping
 from typing import MutableMapping
 from typing import Optional
 from typing import overload
+from typing import Sequence
 from typing import TextIO
 from typing import Tuple
 from typing import TYPE_CHECKING
@@ -97,7 +98,7 @@ def configure(
     tag: Optional[str] = None,
     template_args: Optional[Dict[str, Any]] = None,
     render_as_batch: bool = False,
-    target_metadata: Optional[MetaData] = None,
+    target_metadata: Optional[MetaData | Sequence[MetaData]] = None,
     include_name: Optional[
         Callable[
             [

diff --git a/alembic/runtime/environment.py b/alembic/runtime/environment.py
@@ -10,6 +10,7 @@
 from typing import MutableMapping
 from typing import Optional
 from typing import overload
+from typing import Sequence
 from typing import TextIO
 from typing import Tuple
 from typing import TYPE_CHECKING
@@ -416,7 +417,7 @@ def configure(
         tag: Optional[str] = None,
         template_args: Optional[Dict[str, Any]] = None,
         render_as_batch: bool = False,
-        target_metadata: Optional[MetaData] = None,
+        target_metadata: Optional[MetaData | Sequence[MetaData]] = None,
         include_name: Optional[IncludeNameFn] = None,
         include_object: Optional[IncludeObjectFn] = None,
         include_schemas: bool = False,