diff --git a/stubs/Flask-Migrate/flask_migrate/__init__.pyi b/stubs/Flask-Migrate/flask_migrate/__init__.pyi index a83f7f783..4d7664402 100644 --- a/stubs/Flask-Migrate/flask_migrate/__init__.pyi +++ b/stubs/Flask-Migrate/flask_migrate/__init__.pyi @@ -1,7 +1,7 @@ # pyright: reportInvalidStubStatement=none import sys -from _typeshed import StrPath, SupportsKeysAndGetItem, SupportsWrite +from _typeshed import StrPath, SupportsFlush, SupportsKeysAndGetItem, SupportsWrite from argparse import Namespace from collections.abc import Callable, Iterable, Sequence from logging import Logger @@ -20,9 +20,7 @@ _AlembicConfigValue: TypeAlias = Any alembic_version: tuple[int, int, int] log: Logger -# TODO: Use _typeshed.SupportsFlush when it's available in type checkers. -class _SupportsWriteAndFlush(SupportsWrite[_T_contra], Protocol): - def flush(self) -> object: ... +class _SupportsWriteAndFlush(SupportsWrite[_T_contra], SupportsFlush, Protocol): ... class Config: # should inherit from alembic.config.Config which is not possible yet template_directory: str | None