From: CaselIT <cfederico87@gmail.com>
Date: Tue, 21 Dec 2021 19:17:39 +0000 (+0100)
Subject: fix type errors raised by mypy .920
X-Git-Tag: rel_1_7_6~8
X-Git-Url: http://git.ipfire.org/cgi-bin/gitweb.cgi?a=commitdiff_plain;h=fa931a444b078df031280a580b9768193e45b6fa;p=thirdparty%2Fsqlalchemy%2Falembic.git

fix type errors raised by mypy .920

Change-Id: Ifca989610193074757dec6602af268ea5139c4bc
---

diff --git a/alembic/script/revision.py b/alembic/script/revision.py
index 4b4e29c9..2bfb7f9d 100644
--- a/alembic/script/revision.py
+++ b/alembic/script/revision.py
@@ -7,6 +7,7 @@ from typing import Collection
 from typing import Deque
 from typing import Dict
 from typing import FrozenSet
+from typing import Iterable
 from typing import Iterator
 from typing import List
 from typing import Optional
@@ -111,7 +112,7 @@ class RevisionMap:
 
     """
 
-    def __init__(self, generator: Callable[[], Iterator["Revision"]]) -> None:
+    def __init__(self, generator: Callable[[], Iterable["Revision"]]) -> None:
         """Construct a new :class:`.RevisionMap`.
 
         :param generator: a zero-arg callable that will generate an iterable
@@ -665,7 +666,7 @@ class RevisionMap:
 
     def filter_for_lineage(
         self,
-        targets: Sequence[_T],
+        targets: Iterable[_T],
         check_against: Optional[str],
         include_dependencies: bool = False,
     ) -> Tuple[_T, ...]:
diff --git a/alembic/util/sqla_compat.py b/alembic/util/sqla_compat.py
index 4d0041e1..0f2762ce 100644
--- a/alembic/util/sqla_compat.py
+++ b/alembic/util/sqla_compat.py
@@ -157,9 +157,9 @@ def _get_connection_in_transaction(connection: Optional["Connection"]) -> bool:
 
 def _copy(schema_item: _CE, **kw) -> _CE:
     if hasattr(schema_item, "_copy"):
-        return schema_item._copy(**kw)
+        return schema_item._copy(**kw)  # type: ignore[union-attr]
     else:
-        return schema_item.copy(**kw)
+        return schema_item.copy(**kw)  # type: ignore[union-attr]
 
 
 def _get_connection_transaction(