From e131421d8eb729a6903652dae8d1e2f898072c13 Mon Sep 17 00:00:00 2001
From: Jahongir Qurbonov <=>
Date: Wed, 21 Aug 2024 21:58:20 +0500
Subject: [PATCH] Fix type annotation for button type

---
 client/src/telethon/_impl/client/types/keyboard.py | 14 ++++++++------
 1 file changed, 8 insertions(+), 6 deletions(-)

diff --git a/client/src/telethon/_impl/client/types/keyboard.py b/client/src/telethon/_impl/client/types/keyboard.py
index a9e51e15..26ced13e 100644
--- a/client/src/telethon/_impl/client/types/keyboard.py
+++ b/client/src/telethon/_impl/client/types/keyboard.py
@@ -1,11 +1,13 @@
-from typing import Optional, TypeAlias
+from typing import Generic, Optional, TypeAlias, TypeVar
 
 from ...tl import abcs, types
 from .buttons.button import Button
 
+AnyButton = TypeVar("AnyButton", bound=Button)
+
 
 def _build_keyboard_rows(
-    btns: list[Button] | list[list[Button]],
+    btns: list[AnyButton] | list[list[AnyButton]],
 ) -> list[abcs.KeyboardButtonRow]:
     # list[button] -> list[list[button]]
     # This does allow for "invalid" inputs (mixing lists and non-lists), but that's acceptable.
@@ -21,12 +23,12 @@ def _build_keyboard_rows(
     ]
 
 
-class Keyboard:
+class Keyboard(Generic[AnyButton]):
     __slots__ = ("_raw",)
 
     def __init__(
         self,
-        buttons: list[Button] | list[list[Button]],
+        buttons: list[AnyButton] | list[list[AnyButton]],
         resize: bool,
         single_use: bool,
         selective: bool,
@@ -43,10 +45,10 @@ class Keyboard:
         )
 
 
-class InlineKeyboard:
+class InlineKeyboard(Generic[AnyButton]):
     __slots__ = ("_raw",)
 
-    def __init__(self, buttons: list[Button] | list[list[Button]]) -> None:
+    def __init__(self, buttons: list[AnyButton] | list[list[AnyButton]]) -> None:
         self._raw = types.ReplyInlineMarkup(rows=_build_keyboard_rows(buttons))