diff options
Diffstat (limited to 'lib/sqlalchemy/sql/selectable.py')
-rw-r--r-- | lib/sqlalchemy/sql/selectable.py | 11 |
1 files changed, 5 insertions, 6 deletions
diff --git a/lib/sqlalchemy/sql/selectable.py b/lib/sqlalchemy/sql/selectable.py index f43e6b43f..47cf68357 100644 --- a/lib/sqlalchemy/sql/selectable.py +++ b/lib/sqlalchemy/sql/selectable.py @@ -109,6 +109,7 @@ if TYPE_CHECKING: from ._typing import _ColumnExpressionOrStrLabelArgument from ._typing import _FromClauseArgument from ._typing import _JoinTargetArgument + from ._typing import _LimitOffsetType from ._typing import _MAYBE_ENTITY from ._typing import _NOT_ENTITY from ._typing import _OnClauseArgument @@ -3955,7 +3956,7 @@ class GenerativeSelect(SelectBase, Generative): def _offset_or_limit_clause( self, - element: Union[int, _ColumnExpressionArgument[Any]], + element: _LimitOffsetType, name: Optional[str] = None, type_: Optional[_TypeEngineArgument[int]] = None, ) -> ColumnElement[Any]: @@ -4041,8 +4042,7 @@ class GenerativeSelect(SelectBase, Generative): @_generative def limit( - self: SelfGenerativeSelect, - limit: Union[int, _ColumnExpressionArgument[int]], + self: SelfGenerativeSelect, limit: _LimitOffsetType ) -> SelfGenerativeSelect: """Return a new selectable with the given LIMIT criterion applied. @@ -4078,7 +4078,7 @@ class GenerativeSelect(SelectBase, Generative): @_generative def fetch( self: SelfGenerativeSelect, - count: Union[int, _ColumnExpressionArgument[int]], + count: _LimitOffsetType, with_ties: bool = False, percent: bool = False, ) -> SelfGenerativeSelect: @@ -4133,8 +4133,7 @@ class GenerativeSelect(SelectBase, Generative): @_generative def offset( - self: SelfGenerativeSelect, - offset: Union[int, _ColumnExpressionArgument[int]], + self: SelfGenerativeSelect, offset: _LimitOffsetType ) -> SelfGenerativeSelect: """Return a new selectable with the given OFFSET criterion applied. |