Expose the new InputMedia types

delivrance · delivrance · commit 92fdf79d5451 · 2018-08-01T21:19:11.000+02:00
diff --git a/pyrogram/__init__.py b/pyrogram/__init__.py
@@ -28,11 +28,10 @@
 from .api.errors import Error
 from .client.types import (
     Audio, Chat, ChatMember, ChatMembers, ChatPhoto, Contact, Document, InputMediaPhoto,
-    InputMediaVideo, InputPhoneContact, Location, Message, MessageEntity,
-    Dialog, Dialogs, Photo, PhotoSize, Sticker, Update, User, UserProfilePhotos,
-    Venue, GIF, Video, VideoNote, Voice, CallbackQuery, Messages, ForceReply,
-    InlineKeyboardButton, InlineKeyboardMarkup, KeyboardButton, ReplyKeyboardMarkup,
-    ReplyKeyboardRemove
+    InputMediaVideo, InputMediaDocument, InputMediaAudio, InputMediaAnimation, InputPhoneContact,
+    Location, Message, MessageEntity, Dialog, Dialogs, Photo, PhotoSize, Sticker, Update, User,
+    UserProfilePhotos, Venue, GIF, Video, VideoNote, Voice, CallbackQuery, Messages, ForceReply,
+    InlineKeyboardButton, InlineKeyboardMarkup, KeyboardButton, ReplyKeyboardMarkup, ReplyKeyboardRemove
 )
 from .client import (
     Client, ChatAction, ParseMode, Emoji,
diff --git a/pyrogram/client/types/__init__.py b/pyrogram/client/types/__init__.py
@@ -23,6 +23,10 @@
 from .chat_photo import ChatPhoto
 from .dialog import Dialog
 from .dialogs import Dialogs
+from .input_media import InputMedia
+from .input_media_animation import InputMediaAnimation
+from .input_media_audio import InputMediaAudio
+from .input_media_document import InputMediaDocument
 from .input_media_photo import InputMediaPhoto
 from .input_media_video import InputMediaVideo
 from .input_phone_contact import InputPhoneContact
