From 76f024c3a5c52a18604dd717c6e43626caebaf10 Mon Sep 17 00:00:00 2001 From: Andrei Morozov Date: Wed, 24 Aug 2022 12:02:39 +0000 Subject: [PATCH] Fix type hints for keyboard builders --- aiogram/utils/keyboard.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/aiogram/utils/keyboard.py b/aiogram/utils/keyboard.py index b28582c9..a5928b07 100644 --- a/aiogram/utils/keyboard.py +++ b/aiogram/utils/keyboard.py @@ -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())