Fix type hints for keyboard builders

This commit is contained in:
Andrei Morozov 2022-08-24 12:02:39 +00:00
parent e7cee430b2
commit 76f024c3a5

View file

@ -238,7 +238,7 @@ class KeyboardBuilder(Generic[ButtonType]):
button = self._button_type(**kwargs)
return self.add(button)
def as_markup(self, **kwargs: Any) -> Union[InlineKeyboardMarkup, ReplyKeyboardMarkup]:
def as_markup(self, **kwargs: Any) -> "ButtonType":
if self._button_type is KeyboardButton:
return ReplyKeyboardMarkup(keyboard=self.export(), **kwargs)
return InlineKeyboardMarkup(inline_keyboard=self.export())