fix type errors raised by mypy .920

Change-Id: Ifca989610193074757dec6602af268ea5139c4bc

diff --git a/alembic/script/revision.py b/alembic/script/revision.py
index 4b4e29c9c7fa5a2fedbb55c41dc55ab2bfd71747..2bfb7f9d44326489ee3341374f15164b20b3076e 100644 (file)
--- 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 4d0041e10cc23181c4cccec9737512a333ad162b..0f2762cea41d92da12f7c8af33e26ecd6da7e1b8 100644 (file)
--- 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(