Update inline_keyboard.py

2020-12-24 15:42:11 +01:00 · 2020-12-24 15:42:11 +01:00 · 16595123c5
parent 6e6ac10674
commit 16595123c5
1 changed files with 2 additions and 1 deletions
--- a/pykeyboard/inline_keyboard.py
+++ b/pykeyboard/inline_keyboard.py
@ -8,8 +8,9 @@ class InlineKeyboard(InlineKeyboardMarkup):
        self.row_width = row_width

    def add(self, *args):
+        checkType = lambda element : True if type(element).__name__ == "InlineKeyboardButton" else False
        self.inline_keyboard = [
-            args[i:i + self.row_width]
+            list(filter(checkType,list(args[i:i + self.row_width])))
            for i in range(0, len(args), self.row_width)
        ]