Fix type annotation for inline button type

This commit is contained in:
Jahongir Qurbonov 2024-08-21 22:15:15 +05:00
parent e131421d8e
commit 3eccbede9f

View File

@ -1,9 +1,10 @@
from typing import Generic, Optional, TypeAlias, TypeVar
from ...tl import abcs, types
from .buttons.button import Button
from .buttons import Button, InlineButton
AnyButton = TypeVar("AnyButton", bound=Button)
AnyInlineButton = TypeVar("AnyInlineButton", bound=InlineButton)
def _build_keyboard_rows(
@ -45,10 +46,12 @@ class Keyboard(Generic[AnyButton]):
)
class InlineKeyboard(Generic[AnyButton]):
class InlineKeyboard(Generic[AnyInlineButton]):
__slots__ = ("_raw",)
def __init__(self, buttons: list[AnyButton] | list[list[AnyButton]]) -> None:
def __init__(
self, buttons: list[AnyInlineButton] | list[list[AnyInlineButton]]
) -> None:
self._raw = types.ReplyInlineMarkup(rows=_build_keyboard_rows(buttons))